Update app.py

app.py

@@ -43,7 +43,7 @@ polite_guard = PoliteGuardTool()
 web_search = DuckDuckGoSearchTool()
 
 # Switch to Gemini if this model gets overloaded
-
+'''
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -51,13 +51,14 @@ model = HfApiModel(
      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")
-#)
+model = LiteLLMModel(
+  model_id="gemini/gemini-2.0-flash-exp",
+  max_tokens=2096,
+ temperature=0.6,
+ api_key=os.getenv("LITELLM_API_KEY")
+)
 
 
 # Import tool from Hub