Skip to content

Commit f55dda3

Browse files
author
Evan Lin
committed
đź”§ Update generative model to 'gemini-2.0-flash-lite' for enhanced performance in text and image generation functions.
1 parent 98a5aee commit f55dda3

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

‎main.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ def generate_gemini_text_complete(prompt):
9898
"""
9999
Generate a text completion using the generative model.
100100
"""
101-
model = genai.GenerativeModel('gemini-1.5-flash')
101+
model = genai.GenerativeModel('gemini-2.0-flash-lite')
102102
response = model.generate_content(prompt)
103103
return response
104104

@@ -108,7 +108,7 @@ def generate_result_from_image(img, prompt):
108108
Generate a image vision result using the generative model.
109109
"""
110110

111-
model = genai.GenerativeModel('gemini-1.5-flash')
111+
model = genai.GenerativeModel('gemini-2.0-flash-lite')
112112
response = model.generate_content([prompt, img], stream=True)
113113
response.resolve()
114114
return response

0 commit comments

Comments
 (0)