Spaces:
Sleeping
Sleeping
nageshsomayajula
commited on
Commit
•
0b04dcb
1
Parent(s):
4cd32d8
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import time
|
|
5 |
|
6 |
|
7 |
|
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
|
|
|
5 |
|
6 |
|
7 |
|
8 |
+
def call_chat(message,history):
|
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
|