Update app.py

app.py

@@ -1,6 +1,6 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 
-# Google gemini
+# Google Gemini
 # from litellm import  LiteLLMModel, RateLimitError
 
 import datetime
@@ -31,9 +31,7 @@ def get_the_current_time_in_timezone(timezone: str) -> str:
         # Create timezone object
         tz = pytz.timezone(timezone)
         # Get current time in that timezone
-        # Just get the time, not the entire date and time
         local_time = datetime.datetime.now(tz).strftime("%I:%M %p")  # %I for 12-hour clock, %M for minutes, %p for AM/PM
-        #local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
         return f"The current local time in {timezone} is: {local_time}"
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
@@ -43,20 +41,21 @@ final_answer = FinalAnswerTool()
 polite_guard = PoliteGuardTool()
 web_search = DuckDuckGoSearchTool()
 
+# Switch to Gemini if this model gets overloaded
 
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
     model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-    # it is possible that this model may be overloaded
      custom_role_conversions=None,
  )
 
+
 #model = LiteLLMModel(
 #  model_id="gemini/gemini-2.0-flash-exp",
 #  max_tokens=2096,
- # temperature=0.6,
- # api_key=os.getenv("LITELLM_API_KEY")
+# temperature=0.6,
+# api_key=os.getenv("LITELLM_API_KEY")
 #)
 
 
@@ -73,7 +72,7 @@ with open("contentprompts.yml", 'r') as stream:
 
 combined_prompts = {**prompt_templates, **content_prompts}
 
-#
+
 #  web_search, visit_webpage 
     
 agent = CodeAgent(