nageshsomayajula commited on
Commit
47eeb32
·
1 Parent(s): 8f74ad8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -8,8 +8,7 @@ import time
8
  def call_chat(message):
9
  llm = AutoModelForCausalLM.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1", model_type="Mistral")
10
  for text in llm(message, stream=True):
11
- response =(text, end="", flush=True)
12
- return response
13
 
14
  with gr.Blocks() as demo:
15
  chatbot = gr.Chatbot()
 
8
  def call_chat(message):
9
  llm = AutoModelForCausalLM.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1", model_type="Mistral")
10
  for text in llm(message, stream=True):
11
+ return text
 
12
 
13
  with gr.Blocks() as demo:
14
  chatbot = gr.Chatbot()