Update app.py

app.py

@@ -1,17 +1,30 @@
 import gradio as gr
-from transformers import pipeline
+import torch
+from transformers import pipeline, logging
 
-# بارگذاری مدل DeepSeek-Coder
-coder = pipeline("text-generation", model="deepseek-ai/deepseek-coder-33b-instruct")
+# کاهش مصرف حافظه
+logging.set_verbosity_error()
+torch.backends.cudnn.benchmark = True
+
+# استفاده از مدل سبک‌تر
+model_name = "deepseek-ai/deepseek-coder-1.3b-base"
+coder = pipeline(
+    "text-generation",
+    model=model_name,
+    device=0 if torch.cuda.is_available() else -1,
+    torch_dtype=torch.float16
+)
 
 def respond(message, history):
-    prompt = f"<|system|>You are an expert programming assistant. Use Persian when needed.</s><|user|>{message}</s><|assistant|>"
-    response = coder(prompt, max_new_tokens=200)[0]['generated_text']
-    return response.split("<|assistant|>")[-1].strip()
+    try:
+        prompt = f"### سوال: {message}\n### پاسخ:"
+        response = coder(prompt, max_new_tokens=150)[0]['generated_text']
+        return response.split("### پاسخ:")[-1].strip()
+    except Exception as e:
+        return f"خطا: {str(e)}"
 
-# استفاده از تمپلت آماده Chat Interface
 gr.ChatInterface(
     respond,
-    title="👨‍💻 دستیار کدنویسی شما",
-    examples=["چطوری در React از useState استفاده کنم؟", "خطای SyntaxError در پایتون را رفع کن"]
-).launch()
+    title="🧑‍💻 دستیار برنامه‌نویسی",
+    description="پرسش‌های برنامه‌نویسی خود را مطرح کنید"
+).launch(server_port=7860, share=False)