Update default model name (#576)

Change-Id: I2d8ef6d251d0128695e45df119c31d6b63e26ba7
google-gemini · Sep 30, 2024 · 4354af6 · 4354af6
1 parent b50341e
commit 4354af6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/google/generativeai/generative_models.py b/google/generativeai/generative_models.py
@@ -72,7 +72,7 @@ class GenerativeModel:
 
     def __init__(
         self,
-        model_name: str = "gemini-pro",
+        model_name: str = "gemini-1.5-flash-002",
         safety_settings: safety_types.SafetySettingOptions | None = None,
         generation_config: generation_types.GenerationConfigType | None = None,
         tools: content_types.FunctionLibraryType | None = None,