Final_Assignment_Template

Sleeping

dygoo commited on May 3

Commit

c7dbf16

verified ·

1 Parent(s): 1cf6c37

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -138,7 +138,7 @@ class BasicAgent:
         model_name = "gemini-pro"
         if "gemini-2.0" in self.model:
-            model_name = "gemini-1.5-pro"
         self.gemini_model = genai.GenerativeModel(
             model_name=model_name,

         model_name = "gemini-pro"
         if "gemini-2.0" in self.model:
+            model_name = "gemini-2.5-pro-exp-03-25"
         self.gemini_model = genai.GenerativeModel(
             model_name=model_name,