Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files- ai_engine.py +2 -2
ai_engine.py
CHANGED
|
@@ -5,9 +5,9 @@ import json
|
|
| 5 |
# Configure the Gemini API
|
| 6 |
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
|
| 7 |
|
| 8 |
-
# Initialize the model (
|
| 9 |
model = genai.GenerativeModel(
|
| 10 |
-
model_name='gemini-1.5-flash',
|
| 11 |
generation_config={"response_mime_type": "application/json"} # Forces perfect JSON!
|
| 12 |
)
|
| 13 |
|
|
|
|
| 5 |
# Configure the Gemini API
|
| 6 |
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
|
| 7 |
|
| 8 |
+
# Initialize the model (using the standard gemini-pro which is universally supported)
|
| 9 |
model = genai.GenerativeModel(
|
| 10 |
+
model_name='gemini-1.5-flash-latest',
|
| 11 |
generation_config={"response_mime_type": "application/json"} # Forces perfect JSON!
|
| 12 |
)
|
| 13 |
|