Update app.py

app.py

@@ -3,22 +3,24 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
 # تحميل الموديل
-model_name = "deepseek-ai/deepseek-coder-1.3b-base"
+model_name = "Salesforce/codegen-350M-mono"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name).to("cpu")
 
-# دالة توليد الكود
+# دالة التوليد
 def generate_code(prompt):
-    inputs = tokenizer(prompt, return_tensors="pt").to("cpu")
+    inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(
         **inputs,
-        max_new_tokens=200,       # طول الكود الناتج
-        temperature=0.7,          # درجة الإبداع
-        do_sample=True,
-        top_k=50,
-        top_p=0.95
+        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)
+
+    # إزالة البادئة من الـ prompt لو تكررت في الإخراج
     return code[len(prompt):] if code.startswith(prompt) else code
 
 # واجهة Gradio
@@ -26,6 +28,6 @@ gr.Interface(
     fn=generate_code,
     inputs=gr.Textbox(lines=5, placeholder="Describe what code you want...", label="Prompt"),
     outputs=gr.Textbox(label="Generated Code"),
-    title="Code Generator - DeepSeek Model",
-    description="Generate Python code from a text description using DeepSeek Coder 1.3B Base model"
+    title="Code Generator - Mono Model",
+    description="Generate Python code from a text description using CodeGen-350M-Mono model"
 ).launch()