Update app.py

app.py

@@ -33,7 +33,7 @@ parsed_descriptions_queue = deque()
 
 # Usage limits
 MAX_DESCRIPTIONS = 30
-MAX_IMAGES = 1  # Generate only 1 image
+MAX_IMAGES = 2  # Limit to 2 images
 
 # Preload models and checkpoints
 print("Preloading models and checkpoints...")
@@ -133,15 +133,14 @@ def generate_descriptions(user_prompt, seed_words_input, batch_size=100, max_ite
     return list(parsed_descriptions_queue)
 
 @spaces.GPU(duration=120)
-def generate_images(parsed_descriptions, max_iterations=1):  # Set max_iterations to 1
-    if len(parsed_descriptions) < MAX_IMAGES:
-        prompts = parsed_descriptions
-    else:
-        prompts = [parsed_descriptions.pop(0) for _ in range(MAX_IMAGES)]
+def generate_images(parsed_descriptions, max_iterations=2:  # Set max_iterations to 1
+    # Limit the number of descriptions passed to the image generator to 2
+    if len(parsed_descriptions) > MAX_IMAGES:
+        parsed_descriptions = parsed_descriptions[:MAX_IMAGES]
 
     images = []
-    for prompt in prompts:
-        images.extend(pipe(prompt, num_images=1, num_inference_steps=max_iterations, height=512, width=512).images)  # Set resolution to 512 x 512
+    for prompt in parsed_descriptions:
+        images.extend(pipe(prompt, num_inference_steps=max_iterations, height=512, width=512).images)  # Set resolution to 512 x 512
 
     return images