walaa2022 commited on
Commit
07559cf
·
verified ·
1 Parent(s): 0bee0bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def setup_gemini():
26
  raise ValueError("GEMINI_API_KEY not found in secrets")
27
 
28
  genai.configure(api_key=api_key)
29
- return genai.GenerativeModel('gemini-1.5-pro')
30
  except Exception as e:
31
  print(f"Error setting up Gemini: {e}")
32
  return None
 
26
  raise ValueError("GEMINI_API_KEY not found in secrets")
27
 
28
  genai.configure(api_key=api_key)
29
+ return genai.GenerativeModel('gemini-2.0-flash')
30
  except Exception as e:
31
  print(f"Error setting up Gemini: {e}")
32
  return None