Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, WikipediaSearchTool
+from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool, WikipediaSearchTool, HfApiModel
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -12,8 +12,9 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        model = OpenAIServerModel(model_id="gpt-4o",
-                                 api_key=os.getenv("OPENAI_API_KEY"))
+        # model = OpenAIServerModel(model_id="gpt-4o",
+        #                          api_key=os.getenv("OPENAI_API_KEY"))
+        model=HfApiModel(api_key=os.getenv('HUGGING_FACE_API_KEY'))
         # Initialize the DuckDuckGo search tool
         search_tool = DuckDuckGoSearchTool()
         wiki_search = WikipediaSearchTool()