Skip to content

Commit 5e4cae9

Browse files
authored
fix: update gemini sdk (#349)
Update Gemini Model
1 parent d69452a commit 5e4cae9

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/autoslice/mllm_sdk/gemini_old_sdk.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,7 @@ def gemini_generate_title(video_path, artist):
2626
prompt = SLICE_PROMPT.format(artist=artist)
2727

2828
# Set the model to Gemini Flash.
29-
model = genai.GenerativeModel(model_name="models/gemini-2.5-flash")
30-
29+
model = genai.GenerativeModel(model_name="gemini-2.5-flash-preview-05-20")
3130
response = model.generate_content(
3231
[prompt, video_file], request_options={"timeout": 600}
3332
)

0 commit comments

Comments
 (0)