Update app.py

app.py

@@ -53,7 +53,7 @@ def cache_answer(question: str, answer: str):
 
 
 # --- Model Setup ---
-MODEL_NAME = 'mistralai/Mistral-7B-Instruct-v0.2'
+MODEL_NAME = 'deepseek-ai/DeepSeek-R1-Distill-Qwen-7B'#'mistralai/Mistral-7B-Instruct-v0.2'
 
 def load_model():
     """Download and load the model and tokenizer."""
@@ -68,8 +68,8 @@ def load_model():
         raise
 
 # Load the model and tokenizer locally
-#model, tokenizer = load_model()
-model = HfApiModel(model_id='mistralai/Mistral-7B-Instruct-v0.2', max_tokens=512)
+model, tokenizer = load_model()
+#model = HfApiModel(model_id='mistralai/Mistral-7B-Instruct-v0.2', max_tokens=512)
 
 
 class BasicAgent: