Update app.py

app.py

@@ -16,17 +16,18 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
-    def __init__(self):
+    def __init__(self, HF_API_KEY):
         print("BasicAgent initialized.")
         # Initialize a small local model instead of API model
         self.model = HfApiModel("text-generation", model="mistralai/Mistral-7B-v0.1")
-
+        self.HF_API_KEY = HF_API_KEY
         # Initialize the search tool (you can keep DuckDuckGoSearchTool)
         search_tool = DuckDuckGoSearchTool()
 
         # Initialize Agent
         self.agent = CodeAgent(
             model=self.model,
+            api_key = self.HF_API_KEY
             tools=[search_tool]
         )