Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import OpenRouterModel, DuckDuckGoSearchTool, PythonInterpreterTool, WikipediaSearchTool, CodeAgent
+from smolagents import OpenAIServerModel, DuckDuckGoSearchTool, PythonInterpreterTool, WikipediaSearchTool, CodeAgent
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -16,7 +16,7 @@ class BasicAgent:
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
         self.agent = CodeAgent(
-            model=OpenRouterModel(
+            model=OpenAIServerModel(
                 model_id="anthropic:claude-3.7-sonnet",
                 api_key=os.environ["OPENROUTER_API_KEY"]
             ),