Update app.py

app.py

@@ -85,13 +85,20 @@ 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,model = load_model(MODEL_NAME)
+
+from smolagents import TransformersModel
+
+model = TransformersModel(
+    model_id=MODEL_NAME,
+    max_new_tokens=4096)
+
+
 
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         self.agent = CodeAgent(
-            model=llm_engine,
+            model=model,
             tools=[web_search,python_interpreter,visit_webpage_tool], 
             max_steps=1,
             verbosity_level=1,