update code

app.py

@@ -1,17 +1,16 @@
 import os
 import gradio as gr
 
-from mcp.client.stdio import StdioServerParameters
-from smolagents import InferenceClientModel, CodeAgent, ToolCollection
-from smolagents.mcp_client import MCPClient
+from mcp import StdioServerParameters
+from smolagents import InferenceClientModel, CodeAgent, ToolCollection, MCPClient
+
+
 
 
 token = os.getenv("HF_TOKEN")
 if token:
     print(f"Token set: {token[:3]}...")
 
-
-
 try:
     mcp_client = MCPClient(
         ## Try this working example on the hub:
@@ -20,7 +19,7 @@ try:
     )
     tools = mcp_client.get_tools()
 
-    model = InferenceClientModel(provider="hf-inference",api_key=token)
+    model = InferenceClientModel(token=os.getenv("HF_TOKEN"))
     agent = CodeAgent(tools=[*tools], model=model)
 
     demo = gr.ChatInterface(