Update app.py

app.py

@@ -12,12 +12,12 @@ try:
     )
     tools = mcp_client.get_tools()
 
-    model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
-    # model = LiteLLMModel(
-    #     model_id="gemini/gemini-2.5-flash-preview-05-20",  # Gemini model name for LiteLLM
-    #     temperature=0.2,
-    #     api_key=os.environ["GOOGLE_API_KEY"]               # Must have access to Gemini via Vertex AI
-    # )
+    # model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
+    model = LiteLLMModel(
+        model_id="gemini/gemini-2.5-flash-preview-05-20",  # Gemini model name for LiteLLM
+        temperature=0.2,
+        api_key=os.environ["GOOGLE_API_KEY"]               # Must have access to Gemini via Vertex AI
+    )
     agent = CodeAgent(tools=[*tools], model=model)
 
     demo = gr.ChatInterface(