Spaces:
Sleeping
Sleeping
Commit
·
47eeb32
1
Parent(s):
8f74ad8
Update app.py
Browse files
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 |
-
|
| 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()
|