shoom013 commited on
Commit
af4fe01
·
verified ·
1 Parent(s): c616335

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -38,18 +38,17 @@ max_results = 3
38
  min_len = 40
39
  min_distance = 0.35
40
  max_distance = 0.6
41
- temperature = 0.6
42
  max_tokens=5100
43
  top_p=0.85
44
  top_k=1000
45
  frequency_penalty=0.0
46
  repetition_penalty=1.12
47
  presence_penalty=0.15
48
- jezik = "srpski"
49
  cs = "s0"
50
 
51
  system_sr = "Zoveš se U-Chat AI asistent i pomažeš odgovorima korisniku usluga kompanije United Group. Korisnik postavlja pitanje ili problem na koji očekuje rešenje. "
52
- " Ako ne znaš odgovor, reci da ne znaš, ne izmišljaj ga."
53
  system_sr += "Usluge kompanije United Group uključuju i kablovsku mrežu za digitalnu televiziju, pristup internetu, uređaj EON SMART BOX za TV sadržaj, kao i fiksnu telefoniju. "
54
 
55
  chroma_client = chromadb.PersistentClient(CHROMA_DATA_PATH)
@@ -69,7 +68,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
69
  #
70
  login(token=(HF_TOKEN))
71
 
72
- system_prompt = system_sr
73
 
74
  client = InferenceClient(LLM_NAME)
75
 
@@ -115,7 +114,6 @@ def resetChat():
115
  return True
116
 
117
  def rag(input_text, history, jezik): # , file):
118
- global system_prompt
119
  # if (btn):
120
  # resetChat()
121
  # print(history, input_text)
@@ -162,6 +160,7 @@ def rag(input_text, history, jezik): # , file):
162
  # # {"lang": jezik } ] },
163
  #)
164
  #jezik = "N/A"
 
165
  match jezik:
166
  case 'hrvatski':
167
  o_jezik = 'na hrvatskom jeziku, gramatički točno.'
@@ -171,14 +170,14 @@ def rag(input_text, history, jezik): # , file):
171
  system_prompt = system_sr + "Call centar i pomoč za fizične uporabnike: 070 700 700.stran za podporo je <https://telemach.si/pomoc>. " + "Odgovor isključivo " + o_jezik
172
  case 'srpski':
173
  o_jezik = 'na srpskom jeziku, gramatički ispravno.'
174
- ystem_prompt = system_sr + "Call centar telefon je 19900 za sve korisnike. Stranica podrške je <https://sbb.rs/podrska/>. " + "Odgovaraj isključivo " + o_jezik
175
  case 'makedonski':
176
  o_jezik = 'на македонски јазикот граматички точно.'
177
  system_prompt = system_sr + "Stranica podrške je https://mn.nettvplus.com/me/podrska/ za NetTV. " + "Oдговори исклучиво " + o_jezik
178
  case 'Eksperimentalna opcija':
179
  o_jezik = 'N/A'
180
  system_prompt = system_sr + "Call centar telefon je 12755 za Crnu Goru, 0800 31111 za BIH, 070 700 700 u Sloveniji, 19900 u Srbiji, 095 1000 444 za hrvatske korisnike. Odgovori na jeziku istom kao i u postavljenom pitanju ili problemu korisnika."
181
- print("jezik: "+jezik)
182
  system_prompt = system_prompt + " Sledi pitanje ili problem korisnika, sa kojim dalje komuniciraš: "
183
  # if (o_jezik!='N/A'):
184
  # input_text += " - odgovori " + o_jezik + "."
 
38
  min_len = 40
39
  min_distance = 0.35
40
  max_distance = 0.6
41
+ temperature = 0.7
42
  max_tokens=5100
43
  top_p=0.85
44
  top_k=1000
45
  frequency_penalty=0.0
46
  repetition_penalty=1.12
47
  presence_penalty=0.15
 
48
  cs = "s0"
49
 
50
  system_sr = "Zoveš se U-Chat AI asistent i pomažeš odgovorima korisniku usluga kompanije United Group. Korisnik postavlja pitanje ili problem na koji očekuje rešenje. "
51
+ # " Ako ne znaš odgovor, reci da ne znaš, ne izmišljaj ga."
52
  system_sr += "Usluge kompanije United Group uključuju i kablovsku mrežu za digitalnu televiziju, pristup internetu, uređaj EON SMART BOX za TV sadržaj, kao i fiksnu telefoniju. "
53
 
54
  chroma_client = chromadb.PersistentClient(CHROMA_DATA_PATH)
 
68
  #
69
  login(token=(HF_TOKEN))
70
 
71
+ #system_prompt = system_sr
72
 
73
  client = InferenceClient(LLM_NAME)
74
 
 
114
  return True
115
 
116
  def rag(input_text, history, jezik): # , file):
 
117
  # if (btn):
118
  # resetChat()
119
  # print(history, input_text)
 
160
  # # {"lang": jezik } ] },
161
  #)
162
  #jezik = "N/A"
163
+ system_prompt = ""
164
  match jezik:
165
  case 'hrvatski':
166
  o_jezik = 'na hrvatskom jeziku, gramatički točno.'
 
170
  system_prompt = system_sr + "Call centar i pomoč za fizične uporabnike: 070 700 700.stran za podporo je <https://telemach.si/pomoc>. " + "Odgovor isključivo " + o_jezik
171
  case 'srpski':
172
  o_jezik = 'na srpskom jeziku, gramatički ispravno.'
173
+ system_prompt = system_sr + "Call centar telefon je 19900 za sve korisnike. Stranica podrške je <https://sbb.rs/podrska/>. " + "Odgovaraj isključivo " + o_jezik
174
  case 'makedonski':
175
  o_jezik = 'на македонски јазикот граматички точно.'
176
  system_prompt = system_sr + "Stranica podrške je https://mn.nettvplus.com/me/podrska/ za NetTV. " + "Oдговори исклучиво " + o_jezik
177
  case 'Eksperimentalna opcija':
178
  o_jezik = 'N/A'
179
  system_prompt = system_sr + "Call centar telefon je 12755 za Crnu Goru, 0800 31111 za BIH, 070 700 700 u Sloveniji, 19900 u Srbiji, 095 1000 444 za hrvatske korisnike. Odgovori na jeziku istom kao i u postavljenom pitanju ili problemu korisnika."
180
+ print("jezik: "+o_jezik)
181
  system_prompt = system_prompt + " Sledi pitanje ili problem korisnika, sa kojim dalje komuniciraš: "
182
  # if (o_jezik!='N/A'):
183
  # input_text += " - odgovori " + o_jezik + "."