Skip to content

Commit cd88aec

Browse files
author
Evan Lin
committed
đź”§ Update model name to gemini-2.0-flash-lite for improved multimodal performance
1 parent d7fbfff commit cd88aec

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

‎main.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,9 +62,9 @@
6262
parser = WebhookParser(channel_secret)
6363

6464
# Using a single, powerful multimodal model for both text and images.
65-
# gemini-1.5-flash-001 is a powerful, cost-effective model for multimodal tasks.
65+
# gemini-2.0-flash is a powerful, cost-effective model for multimodal tasks.
6666
model = ChatVertexAI(
67-
model_name="gemini-1.5-flash-001",
67+
model_name="gemini-2.0-flash-lite",
6868
project=google_project_id,
6969
location=google_location,
7070
max_output_tokens=2048 # Increased token limit for detailed image descriptions
@@ -198,4 +198,4 @@ def generate_image_description(image_uri):
198198
)
199199

200200
response = model.invoke([message])
201-
return response.content
201+
return response.content

0 commit comments

Comments
 (0)