AminFaraji commited on
Commit
6fc7b17
·
verified ·
1 Parent(s): c72d068

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -226,13 +226,11 @@ def get_llama_response(message: str, history: list) -> str:
226
  #print(template)
227
  chain.prompt=prompt
228
  print(4444444444444444444444444444444444444444444444)
229
- #res = chain.invoke(query_text)
230
- inputs = tokenizer(query_text, return_tensors="pt")
231
- outputs = model.generate(inputs.input_ids)
232
- response = tokenizer.decode(outputs[0], skip_special_tokens=True)
233
  print(6666666666666666666666666666666666666666666666666666666666666666666666666666666666)
234
 
235
- return(response)
236
 
237
  import gradio as gr
238
 
 
226
  #print(template)
227
  chain.prompt=prompt
228
  print(4444444444444444444444444444444444444444444444)
229
+ res = chain.invoke(query_text)
230
+
 
 
231
  print(6666666666666666666666666666666666666666666666666666666666666666666666666666666666)
232
 
233
+ return(res)
234
 
235
  import gradio as gr
236