Upload app.py

app.py

@@ -6,6 +6,11 @@ import pandas as pd
 
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 from smolagents import OpenAIServerModel, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool, InferenceClientModel
+import os
+
+api_key = os.getenv("HF_TOKEN")
+
+
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -15,9 +20,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):
-        openAI_model = OpenAIServerModel(model_id="gpt-4o-mini", api_key="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")
+        #openAI_model = OpenAIServerModel(model_id="gpt-4o-mini", api_key="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")
         self.agent = CodeAgent(
-            model = InferenceClientModel(),
+            model = InferenceClientModel(hf_token=api_key),
             tools = [DuckDuckGoSearchTool(), WikipediaSearchTool()],
             add_base_tools = True,
         )