ahmed792002 commited on
Commit
73f0168
·
verified ·
1 Parent(s): 6409c4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ def chatbot(query, history, system_message, max_length, temperature, top_k, top_
28
 
29
  # Decode generated text
30
  response = tokenizer.decode(final_outputs[0], skip_special_tokens=True)
31
- return response
32
 
33
  # Gradio ChatInterface
34
  demo = gr.ChatInterface(
 
28
 
29
  # Decode generated text
30
  response = tokenizer.decode(final_outputs[0], skip_special_tokens=True)
31
+ return response.split('"')[1]
32
 
33
  # Gradio ChatInterface
34
  demo = gr.ChatInterface(