Update app.py

Add debug prints.

app.py

@@ -22,14 +22,19 @@ class LocalLLM:
     def generate(self, prompt, **kwargs):
         unsupported_keys = ["stop_sequences"] # Remove keys not accepted by HF pipelines
         cleaned_kwargs = {k: v for k, v in kwargs.items() if k not in unsupported_keys}
+        print(f"🧪 kwargs cleaned: {cleaned_kwargs.keys()}")
         try:
             outputs = self.pipe(prompt, **cleaned_kwargs)
             # outputs = [{'generated_text': '...'}]
+            print(f"🧪 Raw output from pipe: {outputs}")
             if isinstance(outputs, list) and isinstance(outputs[0], dict):
-                return outputs[0]["generated_text"]
+                out = outputs[0]["generated_text"]
             elif isinstance(outputs, list):
-                return outputs[0]  # fallback if it's just a list of strings
-            return str(outputs)
+                out = outputs[0]  # fallback if it's just a list of strings
+            else:
+                out = str(outputs)
+            print(f"🧪 Final output string: {out[:100]}")
+            return out
         except Exception as e:
             print(f"❌ Error in LocalLLM.generate(): {e}")
             raise
@@ -281,5 +286,9 @@ if __name__ == "__main__":
 
     print("-"*(60 + len(" App Starting ")) + "\n")
 
+    # Test the agent
+    agent = BasicAgent()
+    print("TEST OUTPUT:", agent.model.generate("What is 2+2?"))
+
     print("Launching Gradio Interface for Basic Agent Evaluation...")
     demo.launch(debug=True, share=False)