Update app.py

app.py

@@ -97,27 +97,24 @@
 #     mcp_client.disconnect()
 
 import gradio as gr
+import os
 
-from mcp.client.stdio import StdioServerParameters
-from smolagents import InferenceClientModel, CodeAgent, ToolCollection
-from smolagents.mcp_client import MCPClient
+from smolagents import InferenceClientModel, CodeAgent, MCPClient
 
 
 try:
     mcp_client = MCPClient(
-        ## Try this working example on the hub:
-        {"url": "https://abidlabs-mcp-tools.hf.space/gradio_api/mcp/sse"}
-        # {"url": "http://localhost:7860/gradio_api/mcp/sse"}
+        {"url": "https://abidlabs-mcp-tool-http.hf.space/gradio_api/mcp/sse"}
     )
     tools = mcp_client.get_tools()
 
-    model = InferenceClientModel()
-    agent = CodeAgent(tools=[*tools], model=model)
+    model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
+    agent = CodeAgent(tools=[*tools], model=model, additional_authorized_imports=["json", "ast", "urllib", "base64"])
 
     demo = gr.ChatInterface(
         fn=lambda message, history: str(agent.run(message)),
         type="messages",
-        examples=["Prime factorization of 68"],
+        examples=["Analyze the sentiment of the following text 'This is awesome'"],
         title="Agent with MCP Tools",
         description="This is a simple agent that uses MCP tools to answer questions.",
     )