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()
|