multimodalart HF Staff commited on
Commit
e5da62c
·
verified ·
1 Parent(s): ba09d13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- for attempt in range(0, 2):
 
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,