nageshsomayajula commited on
Commit
49f17fa
·
1 Parent(s): 0382585

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +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
- return (message, end="", flush=True)
12
 
13
  with gr.Blocks() as demo:
14
  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 (message,flush=True)
12
 
13
  with gr.Blocks() as demo:
14
  chatbot = gr.Chatbot()