Update main.py

main.py

@@ -4,9 +4,8 @@ from pydantic import BaseModel
 
 
 llm = AutoModelForCausalLM.from_pretrained("mistral-7b-instruct-v0.2.Q4_K_M.gguf",
-    model_type='mistral',
+    model_type='mistral-instruct',
     max_new_tokens = 2048,
-    max_context_length = 1024,
     threads = 3,
 )