Update app.py

app.py

@@ -1,7 +1,7 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 
 # Google Gemini
-# from litellm import  LiteLLMModel, RateLimitError
+from litellm import  LiteLLMModel, RateLimitError
 
 import datetime
 import requests
@@ -42,6 +42,7 @@ final_answer = FinalAnswerTool()
 polite_guard = PoliteGuardTool()
 web_search = DuckDuckGoSearchTool()
 
+'''
 # Switch to Gemini if this model gets overloaded
 my_id = 'Qwen/Qwen2.5-Coder-32B-Instruct'
 
@@ -61,7 +62,7 @@ model = LiteLLMModel(
  api_key=os.getenv("LITELLM_API_KEY")
 )
 
-'''
+
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)