Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -70,8 +70,9 @@ def unified_image_generator(
|
|
| 70 |
|
| 71 |
# Always add the prompt to the contents
|
| 72 |
contents.append(prompt)
|
| 73 |
-
|
| 74 |
-
|
|
|
|
| 75 |
try:
|
| 76 |
response = client.models.generate_content(
|
| 77 |
model=GEMINI_MODEL_NAME,
|
|
|
|
| 70 |
|
| 71 |
# Always add the prompt to the contents
|
| 72 |
contents.append(prompt)
|
| 73 |
+
|
| 74 |
+
response = None
|
| 75 |
+
for attempt in range(0, 3):
|
| 76 |
try:
|
| 77 |
response = client.models.generate_content(
|
| 78 |
model=GEMINI_MODEL_NAME,
|