Spaces:
Sleeping
Sleeping
File size: 713 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 26 27 |
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)
print("Listing available Gemini models...")
try:
for m in genai.list_models():
if 'generateContent' in m.supported_generation_methods:
print(m.name)
except Exception as e:
print(f"List models failed: {e}")
model_name = "gemini-1.5-flash"
print(f"\nTesting model: {model_name}")
try:
model = genai.GenerativeModel(model_name)
response = model.generate_content("Hello, can you see this?")
print("Response:", response.text)
except Exception as e:
print(f"Test failed: {e}")
|