Final_Assignment_Template_v2

Sleeping

agentzero07 commited on Sep 3

Commit

c492c56

verified ·

1 Parent(s): eb67b1a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,7 +34,7 @@ class MyAgent:
             raise ValueError("GEMINI_API_KEY not set in environment variables.")
         self.model = LiteLLMModel(
-            model_id="gemini/gemini-2.0-flash-lite",
             api_key=gemini_api_key,
             system_prompt=SYSTEM_PROMPT
         )

             raise ValueError("GEMINI_API_KEY not set in environment variables.")
         self.model = LiteLLMModel(
+            model_id="gemini/gemini-2.5-flash",
             api_key=gemini_api_key,
             system_prompt=SYSTEM_PROMPT
         )