Update app.py

app.py

@@ -77,7 +77,7 @@ def load_model(model_name):
         transformers_engine = TransformersEngine(pipeline("text-generation", model=model, tokenizer=tokenizer))
 
         
-        return transformers_engine
+        return transformers_engine, model
     except Exception as e:
         print(f"Error loading model: {e}")
         raise
@@ -85,12 +85,13 @@ def load_model(model_name):
 # Load the model and tokenizer locally
 #model, tokenizer = load_model()
 #model = HfApiModel()#model_id=MODEL_NAME, max_tokens=512)
-llm_engine = load_model(MODEL_NAME)
+llm_engine,model = load_model(MODEL_NAME)
 
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         self.agent = CodeAgent(
+            model=model,
             llm_engine=llm_engine,
             tools=[web_search,python_interpreter,visit_webpage_tool], 
             max_steps=1,