Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -133,7 +133,7 @@ def generate_descriptions(user_prompt, seed_words_input, batch_size=100, max_ite
|
|
133 |
return list(parsed_descriptions_queue)
|
134 |
|
135 |
@spaces.GPU(duration=120)
|
136 |
-
def generate_images(parsed_descriptions, max_iterations=2: # Set max_iterations to 1
|
137 |
# Limit the number of descriptions passed to the image generator to 2
|
138 |
if len(parsed_descriptions) > MAX_IMAGES:
|
139 |
parsed_descriptions = parsed_descriptions[:MAX_IMAGES]
|
|
|
133 |
return list(parsed_descriptions_queue)
|
134 |
|
135 |
@spaces.GPU(duration=120)
|
136 |
+
def generate_images(parsed_descriptions, max_iterations=2): # Set max_iterations to 1
|
137 |
# Limit the number of descriptions passed to the image generator to 2
|
138 |
if len(parsed_descriptions) > MAX_IMAGES:
|
139 |
parsed_descriptions = parsed_descriptions[:MAX_IMAGES]
|