Update app.py

app.py

@@ -56,7 +56,6 @@ class LoadTextFileTool(Tool):
         with open(file_path, 'r', encoding='utf-8') as file:
             return file.read()
 
-
 prompts = yaml.safe_load(
     importlib.resources.files("smolagents.prompts").joinpath("code_agent.yaml").read_text()
 )
@@ -64,9 +63,8 @@ prompts["system_prompt"] = ("You are a general AI assistant. I will ask you a qu
                             + prompts["system_prompt"])
 
 def init_agent():
-    gemini_model = OpenAIServerModel(
+    model = OpenAIServerModel(
             model_id="gpt-4.1-mini",
-            api_base="https://generativelanguage.googleapis.com/v1beta/openai/",
             api_key=os.getenv("OPENAI_API_KEY"),
             temperature=0.7
     )
@@ -78,9 +76,9 @@ def init_agent():
             GetTaskFileTool(),
             SpeechToTextTool(),
             LoadXlsxFileTool(),
-            LoadTextFileTool()
+            LoadTextFileTool(),
         ],
-        model=gemini_model,
+        model=model,
         prompt_templates=prompts,
         max_steps=15,
         additional_authorized_imports = ["pandas"]