Update app.py

app.py

@@ -227,12 +227,12 @@ class HfAgentWrapper:
             print(f"Prepared Tool objects with names: {[tool.name for tool in self.actual_tools_for_agent]}")
 
             # --- Step 3: Pass the PRE-INITIALIZED pipeline object to HfAgent ---
-            print("Initializing MyCustomHfAgent with the pre-created pipeline...")
-            self.agent = MyCustomHfAgent( # <<< --- USE YOUR NEW CUSTOM CLASS
-                llm_pipeline,
+            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
             )
-            print("MyCustomHfAgent successfully instantiated.")
+            print("HfAgent successfully instantiated with pre-initialized pipeline.")
 
         except Exception as e:
             print(f"CRITICAL ERROR: Failed to initialize HfAgent or Pipeline: {e}")