anaspro commited on
Commit
bd5bd56
·
1 Parent(s): d1b249e
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +6 -15
  3. system_prompt.txt +1 -1
README.md CHANGED
@@ -8,7 +8,7 @@ sdk_version: 5.42.0
8
  app_file: app.py
9
  pinned: false
10
  models:
11
- - anaspro/iraqi-7b
12
  tags:
13
  - customer-support
14
  - arabic
 
8
  app_file: app.py
9
  pinned: false
10
  models:
11
+ - anaspro/iraqi-7b_4bit
12
  tags:
13
  - customer-support
14
  - arabic
app.py CHANGED
@@ -16,15 +16,10 @@ def load_system_prompt():
16
 
17
  DEFAULT_SYSTEM_PROMPT = load_system_prompt()
18
 
19
- model_path = "anaspro/iraqi-7b"
20
 
21
- # Jais chat prompts from documentation
22
- prompt_eng = """### Instruction:Your name is 'Jais', and you are named after Jebel Jais, the highest mountain in UAE. You were made by 'Inception' in the UAE. You are a helpful, respectful, and honest assistant. Always answer as helpfully as possible, while being safe. Complete the conversation between [|Human|] and [|AI|]:
23
- ### Input: [|Human|] {Question}
24
- [|AI|]
25
- ### Response :"""
26
-
27
- prompt_ar = """### Instruction:اسمك "جيس" وسميت على اسم جبل جيس اعلى جبل في الامارات. تم بنائك بواسطة Inception في الإمارات. أنت مساعد مفيد ومحترم وصادق. أجب دائمًا بأكبر قدر ممكن من المساعدة، مع الحفاظ على البقاء أمناً. أكمل المحادثة بين [|Human|] و[|AI|] :
28
  ### Input:[|Human|] {Question}
29
  [|AI|]
30
  ### Response :"""
@@ -110,15 +105,11 @@ def detect_language(text):
110
 
111
  @spaces.GPU()
112
  def generate_response(input_data, chat_history, max_new_tokens, temperature, top_p, top_k, repetition_penalty):
113
- # Detect language of the current question
114
- lang = detect_language(input_data)
115
- prompt_template = prompt_ar if lang == 'ar' else prompt_eng
116
-
117
- # Build conversation for Jais format
118
  conversation_parts = []
119
 
120
- # Add system prompt as part of the instruction (keep it short for Jais)
121
- system_instruction = "اسمك \"أليكس\" وأنت مساعد خدمة العملاء في شركة TechSolutions. مهمتك مساعدة العملاء في حل مشاكلهم مع المنتجات والإجابة عن أسئلتهم حول الخدمات. كن ودوداً وصبوراً ومحترماً. أجب بالعربية أو الإنجليزية حسب تفضيل العميل. ابدأ بالتحية وكن مباشراً في الحلول."
122
 
123
  # Add chat history
124
  if chat_history:
 
16
 
17
  DEFAULT_SYSTEM_PROMPT = load_system_prompt()
18
 
19
+ model_path = "anaspro/iraqi-7b_4bit"
20
 
21
+ # Iraqi Arabic chat prompt
22
+ prompt_ar = """### Instruction:أنت مساعد ذكي تتحدث باللهجة العراقية. أجب دائمًا بأكبر قدر ممكن من المساعدة، مع الحفاظ على البقاء أمناً. أكمل المحادثة بين [|Human|] و[|AI|] :
 
 
 
 
 
23
  ### Input:[|Human|] {Question}
24
  [|AI|]
25
  ### Response :"""
 
105
 
106
  @spaces.GPU()
107
  def generate_response(input_data, chat_history, max_new_tokens, temperature, top_p, top_k, repetition_penalty):
108
+ # Build conversation for Iraqi model format
 
 
 
 
109
  conversation_parts = []
110
 
111
+ # Add system prompt as part of the instruction (Iraqi Arabic)
112
+ system_instruction = "انت مساعد ذكي تتحدث باللهجة العراقية. اسمك أليكس وانت مساعد خدمة العملاء في شركة TechSolutions. مهمتك مساعدة العملاء في حل مشاكلهم مع المنتجات والإجابة عن أسئلتهم حول الخدمات. كن ودود وصبور ومحترم. ابدأ بالتحية وكن مباشر في الحلول."
113
 
114
  # Add chat history
115
  if chat_history:
system_prompt.txt CHANGED
@@ -1 +1 @@
1
- اسمك "أليكس" وأنت مساعد خدمة العملاء في شركة TechSolutions. مهمتك مساعدة العملاء في حل مشاكلهم مع المنتجات والإجابة عن أسئلتهم حول الخدمات. كن ودوداً وصبوراً ومحترماً. أجب بالعربية أو الإنجليزية حسب تفضيل العميل. ابدأ بالتحية وكن مباشراً في الحلول. تحدث باللهجة العراقية البغدادية الطبيعية.
 
1
+ انت