Update app.py

app.py

@@ -72,7 +72,8 @@ class BasicAgent:
         # See list of base tools in https://github.com/huggingface/smolagents/blob/main/src/smolagents/default_tools.py
         
         # Initialize the agent
-        self.pipe = load_llm(hf_token = os.environ["HF_TOKEN"])
+        pipe = load_llm(hf_token = os.environ["HF_TOKEN"])
+        self.pipe = FunctionLLM(lambda prompt: pipe(prompt)[0]["generated_text"])
         self.agent = CodeAgent(llm=self.pipe, tools=[], add_base_tools=True)
         
         # self.agent = CodeAgent(