Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
-import requests
+from transformers import pipeline
 
-# تنظیم کلید API DeepSeek
-DEEPSEEK_API_KEY = "sk-fb523fb5a222445fae65de601818f622"  # کلید API خود را اینجا وارد کنید
-DEEPSEEK_API_URL = "https://api.deepseek.com/v1/chat/completions"  # آدرس API
+# بارگیری مدل از Hugging Face
+model_name = "deepseek-ai/DeepSeek-V2.5-1210"  # نام مدل را اینجا وارد کنید
+pipe = pipeline("text-generation", model=model_name, trust_remote_code=True)
 
 def respond(
     message,
@@ -14,36 +14,23 @@ def respond(
     top_p,
 ):
     # ساخت پیام‌ها برای مدل
-    messages = [{"role": "system", "content": system_message}]
+    messages = [
+        {"role": "system", "content": system_message},
+        {"role": "user", "content": message},
+    ]
 
-    # اضافه کردن تاریخچه چت
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
-
-    messages.append({"role": "user", "content": message})
-
-    # ارسال درخواست به API DeepSeek
-    headers = {
-        "Authorization": f"Bearer {DEEPSEEK_API_KEY}",
-        "Content-Type": "application/json",
-    }
-    payload = {
-        "model": "gpt-3.5-turbo",  # استفاده از مدل gpt-3.5-turbo
-        "messages": messages,
-        "max_tokens": max_tokens,
-        "temperature": temperature,
-        "top_p": top_p,
-    }
+    # تولید متن با مدل
     try:
-        response = requests.post(DEEPSEEK_API_URL, headers=headers, json=payload)
-        response.raise_for_status()  # بررسی خطاهای HTTP
-        response_data = response.json()
-        yield response_data["choices"][0]["message"]["content"]
-    except requests.exceptions.RequestException as e:
-        yield f"خطا در ارتباط با سرور DeepSeek: {str(e)}"
+        response = pipe(
+            messages,
+            max_length=max_tokens,
+            temperature=temperature,
+            top_p=top_p,
+            num_return_sequences=1
+        )
+        yield response[0]['generated_text']
+    except Exception as e:
+        yield f"خطا در تولید متن: {str(e)}"
 
 # ایجاد رابط کاربری با Gradio
 demo = gr.ChatInterface(
@@ -64,5 +51,4 @@ demo = gr.ChatInterface(
 
 # اجرای برنامه
 if __name__ == "__main__":
-    demo.launch()
-    
+    demo.launch()