Update app.py

app.py

@@ -13,7 +13,11 @@ 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 = HfApiModel(model_id="meta-llama/Meta-Llama-3-8B-Instruct", provider="together")
+        model = LiteLLMModel(
+            model_id="gemini/gemini-2.0-flash-lite",
+            api_key= os.environ.get("GEMINI_KEY"),
+            max_tokens=8192
+        )
         self.agent = CodeAgent(
             model = model,
             tools=[GoogleSearchTool()],