Update app.py

app.py

@@ -54,7 +54,7 @@ def cache_answer(question: str, answer: str):
 
 
 # --- Model Setup ---
-MODEL_NAME = 'TinyLlama/TinyLlama-1.1B-Chat-v1.0'#'mistralai/Mistral-7B-Instruct-v0.2'#'microsoft/DialoGPT-small'# 'EleutherAI/gpt-neo-2.7B'#'distilbert/distilgpt2'#'deepseek-ai/DeepSeek-R1-Distill-Qwen-7B'#'mistralai/Mistral-7B-Instruct-v0.2'
+MODEL_NAME = 'meta-llama/Llama-3.1-8B-Instruct'#'TinyLlama/TinyLlama-1.1B-Chat-v1.0'#'mistralai/Mistral-7B-Instruct-v0.2'#'microsoft/DialoGPT-small'# 'EleutherAI/gpt-neo-2.7B'#'distilbert/distilgpt2'#'deepseek-ai/DeepSeek-R1-Distill-Qwen-7B'#'mistralai/Mistral-7B-Instruct-v0.2'
 
 
 
@@ -90,7 +90,7 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         self.agent = CodeAgent(
-            model=llm_engine,
+            llm_engine=llm_engine,
             tools=[web_search,python_interpreter,visit_webpage_tool,model_math_tool], 
             max_steps=1,
             verbosity_level=1,