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ı
|