Update app.py

app.py

@@ -1,12 +1,23 @@
 import gradio as gr
 from transformers import pipeline
 
-# تحميل نموذج دردشة عربي بسيط
-chatbot = pipeline("text-generation", model="aubmindlab/aragpt2-medium", tokenizer="aubmindlab/aragpt2-medium")
+# نستخدم نموذج عربي أخف
+chatbot = pipeline(
+    "text-generation",
+    model="akhooli/gpt2-small-arabic",
+    tokenizer="akhooli/gpt2-small-arabic"
+)
 
 def chat(message, history=[]):
-    prompt = "المستخدم: " + message + "\nالمساعد:"
-    response = chatbot(prompt, max_length=120, num_return_sequences=1, do_sample=True, temperature=0.7)[0]["generated_text"]
+    prompt = f"المحادثة:\nالمستخدم: {message}\nالمساعد:"
+    response = chatbot(
+        prompt,
+        max_length=80,
+        num_return_sequences=1,
+        do_sample=True,
+        temperature=0.7
+    )[0]["generated_text"]
+
     reply = response.split("المساعد:")[-1].strip()
     history.append((message, reply))
     return history, history
@@ -14,8 +25,6 @@ def chat(message, history=[]):
 with gr.Blocks() as demo:
     gr.ChatInterface(
         fn=chat,
-        chatbot=gr.Chatbot(),
-        textbox=gr.Textbox(placeholder="اكتب رسالتك هنا..."),
         title="شات عربي مجاني",
         description="بوت دردشة عربي مبسط يعمل عبر Hugging Face",
         theme="soft"