fixed arena tab compare < 4 not working, fixed height width models not changing always

app.py

@@ -121,7 +121,7 @@ def generate_single_image(
             guidance_scale=decoder_guidance_scale,
         ).images
 
-    # the rest of the models have similar pipeline
+        # the rest of the models have similar pipeline
     else:
         output = pipe(
             prompt=prompt,
@@ -138,7 +138,7 @@ def generate_single_image(
 
 
 # Define the image generation function for the Arena tab
-@spaces.GPU(duration=80)
+@spaces.GPU(duration=200)
 def generate_arena_images(
     prompt,
     negative_prompt,
@@ -189,7 +189,6 @@ def generate_arena_images(
     generator = torch.Generator().manual_seed(seed)
 
     # Generate images for selected models
-    images = []
     if num_models_to_compare >= 2:
         images_a = generate_single_image(
             prompt,
@@ -207,7 +206,6 @@ def generate_arena_images(
             decoder_num_inference_steps_a,
             decoder_guidance_scale_a,
         )
-        images.append(images_a)
         images_b = generate_single_image(
             prompt,
             negative_prompt,
@@ -224,7 +222,9 @@ def generate_arena_images(
             decoder_num_inference_steps_b,
             decoder_guidance_scale_b,
         )
-        images.append(images_b)
+
+        output_arena_images = images_a, images_b, None, None
+        
     if num_models_to_compare >= 3:
         images_c = generate_single_image(
             prompt,
@@ -242,7 +242,9 @@ def generate_arena_images(
             decoder_num_inference_steps_c,
             decoder_guidance_scale_c,
         )
-        images.append(images_c)
+        
+        output_arena_images = images_a, images_b, images_c, None
+
     if num_models_to_compare >= 4:
         images_d = generate_single_image(
             prompt,
@@ -260,9 +262,10 @@ def generate_arena_images(
             decoder_num_inference_steps_d,
             decoder_guidance_scale_d,
         )
-        images.append(images_d)
+        
+        output_arena_images = images_a, images_b, images_c, images_d
 
-    return images
+    return output_arena_images
 
 
 # Define the image generation function for the Individual tab
@@ -998,6 +1001,8 @@ with gr.Blocks(theme=theme, css=css) as demo:
                         prior_guidance_scale_c: gr.update(visible=True),
                         decoder_num_inference_steps_c: gr.update(visible=True),
                         decoder_guidance_scale_c: gr.update(visible=True),
+                        width_c: gr.update(value=1024, maximum=1344),
+                        height_c: gr.update(value=1024, maximum=1344),
                     }
                 elif model_choice_c == "sdxl flash":
                     return {
@@ -1007,6 +1012,8 @@ with gr.Blocks(theme=theme, css=css) as demo:
                         prior_guidance_scale_c: gr.update(visible=False),
                         decoder_num_inference_steps_c: gr.update(visible=False),
                         decoder_guidance_scale_c: gr.update(visible=False),
+                        width_c: gr.update(value=1024, maximum=1344),
+                        height_c: gr.update(value=1024, maximum=1344),
                     }
                 elif model_choice_c == "sd1.5":
                     return {
@@ -1038,8 +1045,8 @@ with gr.Blocks(theme=theme, css=css) as demo:
                         prior_guidance_scale_c: gr.update(visible=False),
                         decoder_num_inference_steps_c: gr.update(visible=False),
                         decoder_guidance_scale_c: gr.update(visible=False),
-                        width_c: gr.update(maximum=1344),
-                        height_c: gr.update(maximum=1344),
+                        width_c: gr.update(value=1024, maximum=1344),
+                        height_c: gr.update(value=1024, maximum=1344),
                     }
 
             def toggle_visibility_arena_d(model_choice_d):
@@ -1051,6 +1058,8 @@ with gr.Blocks(theme=theme, css=css) as demo:
                         prior_guidance_scale_d: gr.update(visible=True),
                         decoder_num_inference_steps_d: gr.update(visible=True),
                         decoder_guidance_scale_d: gr.update(visible=True),
+                        width_d: gr.update(value=1024, maximum=1344),
+                        height_d: gr.update(value=1024, maximum=1344),
                     }
                 elif model_choice_d == "sdxl flash":
                     return {
@@ -1060,6 +1069,8 @@ with gr.Blocks(theme=theme, css=css) as demo:
                         prior_guidance_scale_d: gr.update(visible=False),
                         decoder_num_inference_steps_d: gr.update(visible=False),
                         decoder_guidance_scale_d: gr.update(visible=False),
+                        width_d: gr.update(value=1024, maximum=1344),
+                        height_d: gr.update(value=1024, maximum=1344),
                     }
                 elif model_choice_d == "sd1.5":
                     return {
@@ -1091,8 +1102,8 @@ with gr.Blocks(theme=theme, css=css) as demo:
                         prior_guidance_scale_d: gr.update(visible=False),
                         decoder_num_inference_steps_d: gr.update(visible=False),
                         decoder_guidance_scale_d: gr.update(visible=False),
-                        width_d: gr.update(maximum=1344),
-                        height_d: gr.update(maximum=1344),
+                        width_d: gr.update(value=1024, maximum=1344),
+                        height_d: gr.update(value=1024, maximum=1344),
                     }
 
             model_choice_a.change(
@@ -1426,6 +1437,8 @@ with gr.Blocks(theme=theme, css=css) as demo:
                         prior_guidance_scale: gr.update(visible=True),
                         decoder_num_inference_steps: gr.update(visible=True),
                         decoder_guidance_scale: gr.update(visible=True),
+                        width: gr.update(value=1024, maximum=1344),
+                        height: gr.update(value=1024, maximum=1344),
                     }
                 elif model_choice == "sdxl flash":
                     return {
@@ -1435,6 +1448,8 @@ with gr.Blocks(theme=theme, css=css) as demo:
                         prior_guidance_scale: gr.update(visible=False),
                         decoder_num_inference_steps: gr.update(visible=False),
                         decoder_guidance_scale: gr.update(visible=False),
+                        width: gr.update(value=1024, maximum=1344),
+                        height: gr.update(value=1024, maximum=1344),
                     }
                 elif model_choice == "sd1.5":
                     return {
@@ -1466,8 +1481,8 @@ with gr.Blocks(theme=theme, css=css) as demo:
                         prior_guidance_scale: gr.update(visible=False),
                         decoder_num_inference_steps: gr.update(visible=False),
                         decoder_guidance_scale: gr.update(visible=False),
-                        width: gr.update(maximum=1344),
-                        height: gr.update(maximum=1344),
+                        width: gr.update(value=1024, maximum=1344),
+                        height: gr.update(value=1024, maximum=1344),
                     }
 
             model_choice.change(