Spaces:
Running
Running
pierreguillou
commited on
Commit
•
c0ccd9f
1
Parent(s):
0417125
Update helpers/gemini_functions.py
Browse files
helpers/gemini_functions.py
CHANGED
@@ -16,7 +16,8 @@ def initialize_gemini():
|
|
16 |
"response_mime_type": "text/plain",
|
17 |
}
|
18 |
model = genai.GenerativeModel(
|
19 |
-
model_name="gemini-1.5-pro",
|
|
|
20 |
generation_config=generation_config,
|
21 |
)
|
22 |
return model
|
|
|
16 |
"response_mime_type": "text/plain",
|
17 |
}
|
18 |
model = genai.GenerativeModel(
|
19 |
+
# model_name="gemini-1.5-pro",
|
20 |
+
model_name="gemini-2.0-flash-exp",
|
21 |
generation_config=generation_config,
|
22 |
)
|
23 |
return model
|