BinKhoaLe1812 commited on
Commit
e9b6d16
·
verified ·
1 Parent(s): 7feeadf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -114,11 +114,11 @@ def call_gemini(prompt: str, vision_parts=None) -> str:
114
  else:
115
  kwargs["contents"] = prompt
116
  resp = client.models.generate_content(
117
- model="gemini-2.5-flash-preview-04-17", **kwargs
118
  )
119
  try:
120
  resp = client.models.generate_content(
121
- model="gemini-2.5-flash-preview-06-17", **kwargs
122
  )
123
  # Check for at least one valid candidate
124
  if not resp.candidates:
 
114
  else:
115
  kwargs["contents"] = prompt
116
  resp = client.models.generate_content(
117
+ model="gemini-2.5-pro", **kwargs
118
  )
119
  try:
120
  resp = client.models.generate_content(
121
+ model="gemini-2.5-flash", **kwargs
122
  )
123
  # Check for at least one valid candidate
124
  if not resp.candidates: