Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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,
|