Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -17,6 +17,8 @@ GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
|
|
17 |
if not GEMINI_API_KEY:
|
18 |
raise RuntimeError("GEMINI_API_KEY not found in environment.")
|
19 |
|
|
|
|
|
20 |
def call_gemini(prompt: str) -> str:
|
21 |
"""Use Gemini via REST API instead of gRPC-based SDK"""
|
22 |
url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={GEMINI_API_KEY}"
|
|
|
17 |
if not GEMINI_API_KEY:
|
18 |
raise RuntimeError("GEMINI_API_KEY not found in environment.")
|
19 |
|
20 |
+
model = genai.GenerativeModel("gemini-2.5-flash")
|
21 |
+
|
22 |
def call_gemini(prompt: str) -> str:
|
23 |
"""Use Gemini via REST API instead of gRPC-based SDK"""
|
24 |
url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={GEMINI_API_KEY}"
|