Update app.py

app.py

@@ -15,11 +15,11 @@ model_path= "TheBloke/Mistral-7B-Instruct-v0.2-GGUF"
 
 model = AutoModelForCausalLM.from_pretrained(
     model_path,
-    model_type="llama",
+    model_type="mistral",
 )
 
 def chatbot(message, history):
-    prompt = "USER: " + message + "\nASSISTANT:"
+    prompt = "SYSTEM: You are a helpful assistant\nUSER: " + message + "\nASSISTANT:"
     airemember =  ""
     for human,assistant in history:
         #aimemory.append("USER: " + human + "\nASSISTANT:" + assistant+"\n\n")