pierreguillou commited on
Commit
c0ccd9f
1 Parent(s): 0417125

Update helpers/gemini_functions.py

Browse files
Files changed (1) hide show
  1. helpers/gemini_functions.py +2 -1
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