Spaces:
Sleeping
Sleeping
File size: 638 Bytes
842ca78 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import os
import google.generativeai as genai
from dotenv import load_dotenv
load_dotenv()
api_key = os.getenv("GEMINI_API_KEY") or os.getenv("GOOGLE_API_KEY")
genai.configure(api_key=api_key)
candidates = [
"gemini-2.0-flash",
"gemini-2.0-flash-exp",
"models/gemini-2.0-flash"
]
for model_name in candidates:
print(f"\nTesting model: {model_name}")
try:
model = genai.GenerativeModel(model_name)
response = model.generate_content("Hello")
print(f"β
Success with {model_name}: {response.text}")
break
except Exception as e:
print(f"β Failed with {model_name}: {e}")
|