Update app.py

app.py

@@ -1,22 +1,34 @@
 from transformers import pipeline
 import gradio as gr
+import torch
 
-# تحميل النموذج
-pipe = pipeline("text-generation", model="wasmdashai/Seed-Coder-8B-Instruct-V1")
 
-# دالة توليد الردود
-def chat_with_model(user_input):
-    messages = [
-        {"role": "user", "content": user_input},
-    ]
+# تحميل النموذج باستخدام GPU إذا متوفر
+pipe = pipeline(
+    "text-generation",
+    model="wasmdashai/Seed-Coder-8B-Instruct-V1"
+   
+).cuda()
+
+# دالة الرد
+def respond(message, chat_history):
+    messages = [{"role": "user", "content": m[0]} for m in chat_history]
+    messages.append({"role": "user", "content": message})
+
     output = pipe(messages, max_new_tokens=200, do_sample=True)
-    return output[0]['generated_text']
+    reply = output[0]['generated_text']
+    return reply
+
+# واجهة دردشة Gradio
+chat = gr.ChatInterface(
+    fn=respond,
+    title="Seed-Coder Chatbot",
+    description="دردشة مع نموذج Seed-Coder-8B-Instruct باستخدام GPU",
+    chatbot=gr.Chatbot(height=400),
+    textbox=gr.Textbox(placeholder="اكتب رسالتك هنا...", container=False, scale=7),
+    retry_btn="🔁 إعادة",
+    clear_btn="🗑️ مسح",
+)
 
-# واجهة Gradio
-gr.Interface(
-    fn=chat_with_model,
-    inputs=gr.Textbox(lines=2, placeholder="اكتب سؤالك هنا..."),
-    outputs="text",
-    title="Seed-Coder Chat",
-    description="نموذج Seed-Coder للإجابة على الأسئلة باستخدام نموذج توليد النصوص"
-).launch(share=True)
+# تشغيل التطبيق
+chat.launch()