Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -15,11 +15,11 @@ model_path= "TheBloke/Mistral-7B-Instruct-v0.2-GGUF"
|
|
15 |
|
16 |
model = AutoModelForCausalLM.from_pretrained(
|
17 |
model_path,
|
18 |
-
model_type="
|
19 |
)
|
20 |
|
21 |
def chatbot(message, history):
|
22 |
-
prompt = "
|
23 |
airemember = ""
|
24 |
for human,assistant in history:
|
25 |
#aimemory.append("USER: " + human + "\nASSISTANT:" + assistant+"\n\n")
|
|
|
15 |
|
16 |
model = AutoModelForCausalLM.from_pretrained(
|
17 |
model_path,
|
18 |
+
model_type="mistral",
|
19 |
)
|
20 |
|
21 |
def chatbot(message, history):
|
22 |
+
prompt = "SYSTEM: You are a helpful assistant\nUSER: " + message + "\nASSISTANT:"
|
23 |
airemember = ""
|
24 |
for human,assistant in history:
|
25 |
#aimemory.append("USER: " + human + "\nASSISTANT:" + assistant+"\n\n")
|