riokorb commited on
Commit
0fd701c
·
verified ·
1 Parent(s): 04b9eba

Update app.py to fix gemini API issue

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ class BasicAgent:
47
  raise ValueError("GOOGLE_API_KEY not found in environment variables")
48
 
49
  return Gemini(
50
- model_name="models/gemini-pro",
51
  api_key=api_key,
52
  temperature=0.1,
53
  top_p=0.95,
 
47
  raise ValueError("GOOGLE_API_KEY not found in environment variables")
48
 
49
  return Gemini(
50
+ model_name="models/gemini-1.5-flash",
51
  api_key=api_key,
52
  temperature=0.1,
53
  top_p=0.95,