Update app.py

app.py

@@ -10,7 +10,6 @@ load_dotenv()  # Load environment variables from .env file
 
 import gradio as gr
 from smolagents import InferenceClientModel, CodeAgent, MCPClient
-from llama_index.llms.nebius import NebiusLLM
 
 try:
     mcp_client = MCPClient(
@@ -22,16 +21,7 @@ try:
 
     tools = mcp_client.get_tools()
 
-    NEBIUS_API_KEY = os.getenv("NEBIUS_API_KEY")
-    AGENT_MODEL = os.getenv("AGENT_MODEL", "meta-llama/Meta-Llama-3.1-8B-Instruct")
-
-    # Load an LLM
-    model = NebiusLLM(
-        api_key=NEBIUS_API_KEY,
-        model=AGENT_MODEL,
-        is_function_calling_model=True
-    )
-
+    model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
     agent = CodeAgent(tools=[*tools], model=model)
 
     mcp_description = """