MostlyKIGuess commited on
Commit
c4a80a5
·
1 Parent(s): 2e31ab2

updated model to 2.5 pro

Browse files
Files changed (2) hide show
  1. src/api/fallback_gemini.py +1 -1
  2. src/api/gemini.py +1 -1
src/api/fallback_gemini.py CHANGED
@@ -118,7 +118,7 @@ Provide your response in exactly this format:
118
  )
119
 
120
  response = client.models.generate_content(
121
- model="gemini-2.5-flash",
122
  contents=contents, # type: ignore
123
  config=generation_config,
124
  )
 
118
  )
119
 
120
  response = client.models.generate_content(
121
+ model="gemini-2.5-pro",
122
  contents=contents, # type: ignore
123
  config=generation_config,
124
  )
src/api/gemini.py CHANGED
@@ -175,7 +175,7 @@ def generate_video(idea: str | None = None, pdf_path: str | None = None):
175
  )
176
 
177
  response = client.models.generate_content(
178
- model="gemini-2.5-flash", contents=contents, config=generation_config
179
  )
180
  except Exception as e:
181
  logging.exception(f"Error calling Gemini API: {e}")
 
175
  )
176
 
177
  response = client.models.generate_content(
178
+ model="gemini-2.5-pro", contents=contents, config=generation_config
179
  )
180
  except Exception as e:
181
  logging.exception(f"Error calling Gemini API: {e}")