Update app.py

app.py

@@ -27,7 +27,8 @@ def generate_image(api_key, prompt, image_size='landscape_4_3', num_images=1):
             images.append(img)
         return images
     except Exception as e:
-        return [Image.new('RGB', (300, 300), color='red')]  # Return a red image on error
+        print(f"Error: {e}")
+        return [Image.new('RGB', (300, 300), color='red') for _ in range(num_images)]  # Return red images on error
 
 with gr.Blocks() as demo:
     gr.Markdown("# FLUX1.1 [pro] Text-to-Image Generator")
@@ -43,7 +44,7 @@ with gr.Blocks() as demo:
         )
         num_images = gr.Slider(label="Number of Images", minimum=1, maximum=4, step=1, value=1)
     generate_btn = gr.Button("Generate Image")
-    output_gallery = gr.Gallery(label="Generated Images").style(columns=2, rows=2, height="auto")
+    output_gallery = gr.Gallery(label="Generated Images", columns=2, rows=2)
 
     generate_btn.click(
         fn=generate_image,