Test with gallery on all but main model

app.py

@@ -51,11 +51,11 @@ def run_xlnc(prompt, negative_prompt=None, guidance_scale=7.0, pag_scale=3.0, pa
         seed = random.randint(0, 9007199254740991)
     
     generator = torch.Generator(device="cuda").manual_seed(seed)
-    image = pipe_xlnc(prompt, negative_prompt=negative_prompt, guidance_scale=guidance_scale, pag_scale=pag_scale, pag_applied_layers=pag_layers, generator=generator, num_inference_steps=25).images[0]    
+    image = pipe_xlnc(prompt, negative_prompt=negative_prompt, guidance_scale=guidance_scale, pag_scale=pag_scale, pag_applied_layers=pag_layers, generator=generator, num_inference_steps=25, num_images_per_prompt=4).images
     
     return image, seed
 
-@spaces.GPU(duration=15)
+@spaces.GPU(duration=10)
 def run_sc(prompt, negative_prompt=None, guidance_scale=7.0, pag_scale=3.0, pag_layers=["mid"], randomize_seed=True, seed=42, progress=gr.Progress(track_tqdm=True)):
     if(randomize_seed):
         seed = random.randint(0, 9007199254740991)
@@ -65,7 +65,7 @@ def run_sc(prompt, negative_prompt=None, guidance_scale=7.0, pag_scale=3.0, pag_
     
     return image, seed
 
-@spaces.GPU(duration=15)
+@spaces.GPU(duration=10)
 def run_snc(prompt, negative_prompt=None, guidance_scale=7.0, pag_scale=3.0, pag_layers=["mid"], randomize_seed=True, seed=42, progress=gr.Progress(track_tqdm=True)):
     if(randomize_seed):
         seed = random.randint(0, 9007199254740991)
@@ -114,7 +114,7 @@ with gr.Blocks(css=css, theme=theme) as demo:
       with gr.Row():            
         prompt_sc = gr.Textbox(show_label=False, scale=4, placeholder="Your prompt for S-C")
         button_sc = gr.Button("Generate", min_width=120)
-    output = gr.Image(label="Your result", interactive=False)
+    output = gr.Gallery(label="Your result", interactive=False)
     with gr.Accordion("Advanced Settings", open=False):
         guidance_scale = gr.Number(label="CFG Guidance Scale", info="The guidance scale for CFG, ignored if no prompt is entered (unconditional generation)", value=7.0)
         negative_prompt = gr.Textbox(label="Negative prompt", info="Is only applied for the CFG part, leave blank for unconditional generation")