anaspro commited on
Commit
431107d
·
1 Parent(s): 0e743d8
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  from transformers import pipeline, TextIteratorStreamer
2
  from threading import Thread
3
  import gradio as gr
@@ -13,13 +14,17 @@ def load_system_prompt():
13
 
14
  DEFAULT_SYSTEM_PROMPT = load_system_prompt()
15
 
16
- model_path = "anaspro/model"
 
 
 
17
 
18
  pipe = pipeline(
19
  "text-generation",
20
  model=model_path,
21
  torch_dtype="auto",
22
  device_map="auto",
 
23
  )
24
 
25
  def format_conversation_history(chat_history):
@@ -88,7 +93,7 @@ demo = gr.ChatInterface(
88
  ],
89
  cache_examples=False,
90
  type="messages",
91
- title="دعم عملاء TechSolutions - مساعد أليكس (anaspro AI)",
92
  description="""🤖 مساعد خدمة عملاء ذكي لـ TechSolutions
93
 
94
  ✨ المميزات:
@@ -96,7 +101,7 @@ demo = gr.ChatInterface(
96
  - 💬 لهجة محادثة طبيعية
97
  - 🔧 دعم فني واستكشاف الأخطاء
98
  - 📋 معلومات الخدمات والإرشاد
99
- - 🎯 مدعوم بـ موديل anaspro المدرب خصيصاً (Llama 3.1 مع تحسينات)
100
 
101
  احجي مع أليكس لحل مشاكلك التقنية، استفسر عن الخدمات، أو احصل على معلومات المنتجات.""",
102
  fill_height=True,
 
1
+ import os
2
  from transformers import pipeline, TextIteratorStreamer
3
  from threading import Thread
4
  import gradio as gr
 
14
 
15
  DEFAULT_SYSTEM_PROMPT = load_system_prompt()
16
 
17
+ model_path = "anaspro/meta-llama-3.1-8b-inst-iraqi"
18
+
19
+ # إذا كان فيه HF_TOKEN في البيئة
20
+ hf_token = os.getenv("HF_TOKEN")
21
 
22
  pipe = pipeline(
23
  "text-generation",
24
  model=model_path,
25
  torch_dtype="auto",
26
  device_map="auto",
27
+ token=hf_token, # إضافة التوكن إذا موجود
28
  )
29
 
30
  def format_conversation_history(chat_history):
 
93
  ],
94
  cache_examples=False,
95
  type="messages",
96
+ title="دعم عملاء TechSolutions - مساعد أليكس (العراقي)",
97
  description="""🤖 مساعد خدمة عملاء ذكي لـ TechSolutions
98
 
99
  ✨ المميزات:
 
101
  - 💬 لهجة محادثة طبيعية
102
  - 🔧 دعم فني واستكشاف الأخطاء
103
  - 📋 معلومات الخدمات والإرشاد
104
+ - 🎯 مدعوم بـ موديل anaspro العراقي (Llama 3.1 محسن للعربية العراقية)
105
 
106
  احجي مع أليكس لحل مشاكلك التقنية، استفسر عن الخدمات، أو احصل على معلومات المنتجات.""",
107
  fill_height=True,