丁云鹏 3 months ago
parent
commit
ed1da3fa68
1 changed files with 1 additions and 1 deletions
  1. 1 1
      utils/google_ai_studio.py

+ 1 - 1
utils/google_ai_studio.py

@@ -85,7 +85,7 @@ class GoogleAI(object):
                 os.remove(video_path)
                 return "[异常] 上传视频失败"
             model = genai.GenerativeModel(
-                model_name='gemini-1.5-flash',
+                model_name='gemini-2.0-flash',
                 generation_config=genai.GenerationConfig(response_mime_type='application/json'),
                 safety_settings={
                     HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,