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}"
|