shoom013 commited on
Commit
d8a0832
1 Parent(s): f8e268e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -66,10 +66,10 @@ system_propmpt = system_sr
66
  Settings.llm = HuggingFaceInferenceAPI(model_name="mistralai/Mistral-Nemo-Instruct-2407",
67
  device_map="auto",
68
  system_prompt = system_propmpt,
69
- context_window=4096,
70
- max_new_tokens=512,
71
  # stopping_ids=[50278, 50279, 50277, 1, 0],
72
- generate_kwargs={"temperature": 0.6, "do_sample": False},
73
  # tokenizer_kwargs={"max_length": 4096},
74
  tokenizer_name="mistralai/Mistral-Nemo-Instruct-2407",
75
  )
@@ -160,9 +160,9 @@ def rag(input_text, history, jezik, file):
160
  # outputs=[gr.Textbox(label="Odgovor:", lines=6)],
161
  # ChatMessage(role="assistant", content="Kako Vam mogu pomoći?")
162
  iface = gr.ChatInterface(rag,
163
- title="Agent podrške",
164
  description="Postavite pitanje ili opišite problem koji imate",
165
- chatbot=gr.Chatbot(placeholder="Kako Vam mogu pomoći?", type="tuples", label="UChat", height=300),
166
  textbox=gr.Textbox(placeholder="Pitanje ili opis problema", container=False, scale=7),
167
  theme="soft",
168
  # examples=["Ne radi mi internet", "Koje usluge imam na raspologanju?", "Ne radi mi daljinski upravljač, šta da radim?"],
 
66
  Settings.llm = HuggingFaceInferenceAPI(model_name="mistralai/Mistral-Nemo-Instruct-2407",
67
  device_map="auto",
68
  system_prompt = system_propmpt,
69
+ context_window=5000,
70
+ max_new_tokens=2048,
71
  # stopping_ids=[50278, 50279, 50277, 1, 0],
72
+ generate_kwargs={"temperature": 0.5, "top_p":0.92, "do_sample": False},
73
  # tokenizer_kwargs={"max_length": 4096},
74
  tokenizer_name="mistralai/Mistral-Nemo-Instruct-2407",
75
  )
 
160
  # outputs=[gr.Textbox(label="Odgovor:", lines=6)],
161
  # ChatMessage(role="assistant", content="Kako Vam mogu pomoći?")
162
  iface = gr.ChatInterface(rag,
163
+ title="UChat",
164
  description="Postavite pitanje ili opišite problem koji imate",
165
+ chatbot=gr.Chatbot(placeholder="Kako Vam mogu pomoći?", type="tuples", label="Agent podrške", height=300),
166
  textbox=gr.Textbox(placeholder="Pitanje ili opis problema", container=False, scale=7),
167
  theme="soft",
168
  # examples=["Ne radi mi internet", "Koje usluge imam na raspologanju?", "Ne radi mi daljinski upravljač, šta da radim?"],