Inan Ince
commited on
Commit
·
eb58738
1
Parent(s):
a30219c
Add application file17
Browse files
app.py
CHANGED
|
@@ -8,13 +8,15 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
|
| 8 |
|
| 9 |
# Kod üretme fonksiyonu
|
| 10 |
def generate_java_code(prompt):
|
| 11 |
-
# Girdi metnini
|
|
|
|
|
|
|
|
|
|
| 12 |
inputs = tokenizer(prompt, return_tensors="pt", max_length=512, truncation=True)
|
| 13 |
-
# Modelden kod üret
|
| 14 |
outputs = model.generate(
|
| 15 |
inputs["input_ids"],
|
| 16 |
max_length=300, # Çıktının uzunluğunu artır
|
| 17 |
-
num_beams=5, #
|
| 18 |
early_stopping=True
|
| 19 |
)
|
| 20 |
# Tokenize edilmiş çıktıyı kod olarak döndür
|
|
@@ -23,6 +25,9 @@ def generate_java_code(prompt):
|
|
| 23 |
# Post-processing: Eksik yapıları düzelt
|
| 24 |
if "public static void main" not in code:
|
| 25 |
code = f"public class Main {{\n{code}\n}}"
|
|
|
|
|
|
|
|
|
|
| 26 |
return code
|
| 27 |
|
| 28 |
# Gradio UI tasarımı
|
|
|
|
| 8 |
|
| 9 |
# Kod üretme fonksiyonu
|
| 10 |
def generate_java_code(prompt):
|
| 11 |
+
# Girdi metnini modele uygun bir şekilde biçimlendir
|
| 12 |
+
prompt = f"Generate a complete Java program:\n{prompt}"
|
| 13 |
+
|
| 14 |
+
# Girdiyi tokenize et ve modele gönder
|
| 15 |
inputs = tokenizer(prompt, return_tensors="pt", max_length=512, truncation=True)
|
|
|
|
| 16 |
outputs = model.generate(
|
| 17 |
inputs["input_ids"],
|
| 18 |
max_length=300, # Çıktının uzunluğunu artır
|
| 19 |
+
num_beams=5, # Beam search ile kaliteyi artır
|
| 20 |
early_stopping=True
|
| 21 |
)
|
| 22 |
# Tokenize edilmiş çıktıyı kod olarak döndür
|
|
|
|
| 25 |
# Post-processing: Eksik yapıları düzelt
|
| 26 |
if "public static void main" not in code:
|
| 27 |
code = f"public class Main {{\n{code}\n}}"
|
| 28 |
+
if code.strip().endswith("{"):
|
| 29 |
+
code += "\n // Complete the logic here\n}"
|
| 30 |
+
|
| 31 |
return code
|
| 32 |
|
| 33 |
# Gradio UI tasarımı
|