Update app.py

app.py

@@ -5,14 +5,23 @@ import torch
 # تحميل الموديل
 model_name = "Salesforce/codegen-350M-mono"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name).to("cpu") 
+model = AutoModelForCausalLM.from_pretrained(model_name).to("cpu")
 
 # دالة التوليد
 def generate_code(prompt):
     inputs = tokenizer(prompt, return_tensors="pt")
-    outputs = model.generate(**inputs, max_new_tokens=50)
+    outputs = model.generate(
+        **inputs,
+        max_new_tokens=200,            # زيادة عدد التوكنز ليرجع كود أطول
+        temperature=0.7,               # ضبط العشوائية (أقل من 1 = كود أدق)
+        do_sample=True,                # تفعيل التوليد العشوائي
+        top_k=50,                      # استخدام top-k sampling
+        top_p=0.95                     # أو استخدام nucleus sampling
+    )
     code = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return code
+
+    # إزالة البادئة من الـ prompt لو تكررت في الإخراج
+    return code[len(prompt):] if code.startswith(prompt) else code
 
 # واجهة Gradio
 gr.Interface(