Update app.py

app.py

@@ -20,8 +20,9 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 #model_id="Qwen/Qwen2.5-Coder-32B-Instruct"
 #model=LiteLLMModel(model_id="deepseek-ai/DeepSeek-Coder-V2-Lite-Base", api_key=os.getenv('HF_API_KEY'))
 #model=LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv('GEMINI_API_KEY'))
+model=LiteLLMModel(model_id="gemini/gemini-2.0-flash", api_key=os.getenv('GEMINI_API_KEY'))
 #model=OpenAIServerModel(model_id="gpt-4o", api_key=os.getenv('OPENAI_API_KEY'))
-model=OpenAIServerModel(model_id="gpt-3.5-turbo", api_key=os.getenv('OPENAI_API_KEY'))
+#model=OpenAIServerModel(model_id="gpt-3.5-turbo", api_key=os.getenv('OPENAI_API_KEY'))
 #model=LiteLLMModel(model_id="deepseek/deepseek-chat", api_key=os.getenv('DEEPSEEK_API_KEY'))
 
 # Initialize the model 
@@ -41,19 +42,19 @@ class BasicAgent:
                 #DuckDuckGoSearchTool()
             ],
             model=model,
-            #additional_authorized_imports=[
-            #    "geopandas",
-            #    "plotly",
-            #    "shapely",
-            #    "json",
-            #    "pandas",
-            #    "numpy",
-            #],
+            additional_authorized_imports=[
+                "geopandas",
+                "plotly",
+                "shapely",
+                "json",
+                "pandas",
+                "numpy",
+            ],
             name="agent",
             description="agent desc",
             #planning_interval=5,
             verbosity_level=2,
-            #max_steps=15,
+            max_steps=15,
         )
         
         print("BasicAgent initialized.")