Update app.py

app.py

@@ -1,4 +1,4 @@
-from huggingface_hub import login
+import os
 from smolagents import (
     CodeAgent,
     ToolCallingAgent,
@@ -12,8 +12,12 @@ from markdownify import markdownify
 from requests.exceptions import RequestException
 from transformers import pipeline
 
-# Login to Hugging Face Hub
-login()
+# Set the Hugging Face token using an environment variable
+# Make sure to set the HUGGING_FACE_HUB_TOKEN environment variable in your environment
+# Example: os.environ['HUGGING_FACE_HUB_TOKEN'] = 'your_hugging_face_token'
+
+# Alternatively, you can set it directly in the code for testing purposes
+# os.environ['HUGGING_FACE_HUB_TOKEN'] = 'your_hugging_face_token'
 
 # Define the model ID
 model_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
@@ -138,4 +142,9 @@ btc_agent = CodeAgent(
 
 # Launch the Gradio interface
 from Gradio_UI import GradioUI
-GradioUI(btc_agent).launch()
+
+def gradio_interface(prompt):
+    result = btc_agent.run(prompt)
+    return result
+
+GradioUI(gradio_interface).launch()