Add application file16

app.py

@@ -13,20 +13,31 @@ def generate_java_code(prompt):
     # Modelden kod üret
     outputs = model.generate(
         inputs["input_ids"],
-        max_length=150,  # Çıktı kodunun maksimum uzunluğu
-        num_beams=4,     # Beam search ile daha kaliteli sonuçlar
+        max_length=300,  # Çıktının uzunluğunu artır
+        num_beams=5,     # Daha kaliteli sonuçlar için beam search
         early_stopping=True
     )
     # Tokenize edilmiş çıktıyı kod olarak döndür
     code = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    
+    # Post-processing: Eksik yapıları düzelt
+    if "public static void main" not in code:
+        code = f"public class Main {{\n{code}\n}}"
     return code
 
 # Gradio UI tasarımı
 with gr.Blocks() as demo:
     gr.Markdown("<h1 style='text-align: center;'>Java Kod Üretici (CodeT5)</h1>")
-    prompt = gr.Textbox(label="Doğal Dil Girdisi", placeholder="Örnek: Write a Java program to find the larger of two numbers.")
-    output_code = gr.Textbox(label="Üretilen Java Kodu", lines=10)
+    prompt = gr.Textbox(
+        label="Doğal Dil Girdisi",
+        placeholder="Örnek: Write a Java program to find the larger of two numbers."
+    )
+    output_code = gr.Textbox(
+        label="Üretilen Java Kodu",
+        lines=15,  # Çıktı alanı için daha fazla satır
+    )
     btn = gr.Button("Kod Üret")
     btn.click(generate_java_code, inputs=prompt, outputs=output_code)
 
 demo.launch()
+