Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
| 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 |
-
|
| 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: "+
|
| 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 + "."
|