sikeaditya commited on
Commit
b57de84
·
verified ·
1 Parent(s): 41f3e48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def test_gemini_api():
22
  genai.configure(api_key=api_key)
23
 
24
  # Test with a simple text prompt
25
- model = genai.GenerativeModel('gemini-1.0-pro')
26
  response = model.generate_content("Hello, please respond with 'API is working'")
27
 
28
  if not response or not hasattr(response, 'text') or not response.text:
 
22
  genai.configure(api_key=api_key)
23
 
24
  # Test with a simple text prompt
25
+ model = genai.GenerativeModel('gemini-2.0-flash')
26
  response = model.generate_content("Hello, please respond with 'API is working'")
27
 
28
  if not response or not hasattr(response, 'text') or not response.text: