Update app.py

app.py

@@ -22,18 +22,17 @@ 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()}")
+        # print(f"🧪 kwargs cleaned: {cleaned_kwargs.keys()}")
         try:
             outputs = self.pipe(prompt, **cleaned_kwargs)
-            # outputs = [{'generated_text': '...'}]
-            print(f"🧪 Raw output from pipe: {outputs}")
+            # print(f"🧪 Raw output from pipe: {outputs}")
             if isinstance(outputs, list) and isinstance(outputs[0], dict):
                 out = outputs[0]["generated_text"]
             elif isinstance(outputs, list):
                 out = outputs[0]  # fallback if it's just a list of strings
             else:
                 out = str(outputs)
-            print(f"🧪 Final output string: {out[:100]}")
+            # print(f"🧪 Final output string: {out[:100]}")
             return out
         except Exception as e:
             print(f"❌ Error in LocalLLM.generate(): {e}")
@@ -71,7 +70,7 @@ def check_token_access():
         print("❌ Token check failed:", e)
 
 # --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+# ----- THIS IS WHERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
@@ -286,9 +285,5 @@ 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)