Update app.py

app.py

@@ -3,7 +3,11 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool
+from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, AzureOpenAIModel
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -13,7 +17,12 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-        self.model = InferenceClientModel()
+        self.model = AzureOpenAIModel(
+                        model_id = os.environ.get("AZURE_OPENAI_MODEL"),
+                        azure_endpoint=os.environ.get("AZURE_OPENAI_ENDPOINT"),
+                        api_key=os.environ.get("AZURE_OPENAI_API_KEY"),
+                        api_version=os.environ.get("OPENAI_API_VERSION")
+                    )
         self.agent = CodeAgent(
             tools=[DuckDuckGoSearchTool()], 
             model=self.model,