Update app.py

app.py

@@ -5,12 +5,15 @@ from smolagents import CodeAgent, InferenceClientModel
 from smolagents import DuckDuckGoSearchTool, Tool
 from diffusers import DiffusionPipeline
 import torch
+from smolagents import OpenAIServerModel
 
 from huggingface_hub import login
 import os
 
 token = os.environ.get("HF_TOKEN")
 print(token)
+
+openai_token=  os.environ.get("OPENAI_API_KEY")
 if token:
     login(token=token)
 else:
@@ -83,6 +86,15 @@ search_tool = DuckDuckGoSearchTool()
 #llm_engine = InferenceClientModel("Qwen/Qwen2.5-72B-Instruct")
 
 llm_engine = InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct")
+
+# Inicialização do modelo OpenAI com smolagents
+llm_engine = OpenAIServerModel(
+    model_id="gpt-4o-mini",   # Exemplo: ajuste para o modelo OpenAI que deseja usar
+    api_base="https://api.openai.com/v1",
+    api_key=openai_token
+)
+
+
 agent = CodeAgent(tools=[image_generation_tool, search_tool], model=llm_engine)
 
 # =========================================================