Update app.py

app.py

@@ -228,10 +228,7 @@ class HfAgentWrapper:
 
             # --- Step 3: Pass the PRE-INITIALIZED pipeline object to HfAgent ---
             print("Initializing HfAgent with the pre-created pipeline...")
-            self.agent = HfAgent(
-                llm_pipeline,  # Pass the pipeline object directly as the first argument
-                additional_tools=self.actual_tools_for_agent
-            )
+            self.agent = MyCustomHfAgent(llm_pipeline, additional_tools=self.actual_tools_for_agent)
             print("HfAgent successfully instantiated with pre-initialized pipeline.")
 
         except Exception as e: