Update app.py

app.py

@@ -73,7 +73,7 @@ def generate_(prompt, negative_prompt, normal_image, num_steps, controlnet_condi
         image=normal_image, 
         num_inference_steps=num_steps, 
         controlnet_conditioning_scale=float(controlnet_conditioning_scale),
-        num_images_per_prompt=4,
+        num_images_per_prompt=2,
         generator=generator,
     ).images
     return images
@@ -85,7 +85,7 @@ def process(normal_image, prompt, negative_prompt, num_steps, controlnet_conditi
     # depth_image = get_depth_map(input_image)
     images = generate_(prompt, negative_prompt, normal_image, num_steps, controlnet_conditioning_scale, seed)
 
-    return [images[0], images[1], images[2], images[3]]
+    return [images[0], images[1]]
 
 
 def run_demo():
@@ -115,13 +115,13 @@ def run_demo():
             
                 prompt = gr.Textbox(label="Prompt")
                 negative_prompt = gr.Textbox(visible=False, label="Negative prompt", value="Logo,Watermark,Text,Ugly,Morbid,Extra fingers,Poorly drawn hands,Mutation,Blurry,Extra limbs,Gross proportions,Missing arms,Mutated hands,Long neck,Duplicate,Mutilated,Mutilated hands,Poorly drawn face,Deformed,Bad anatomy,Cloned face,Malformed limbs,Missing legs,Too many fingers")
-                num_steps = gr.Slider(label="Number of steps", minimum=25, maximum=100, value=50, step=1)
+                num_steps = gr.Slider(label="Number of steps", minimum=25, maximum=50, value=30, step=1)
                 controlnet_conditioning_scale = gr.Slider(label="ControlNet conditioning scale", minimum=0.1, maximum=1.0, value=0.95, step=0.05)
                 seed = gr.Slider(label="Seed", minimum=0, maximum=2147483647, step=1, randomize=True,)
                 run_button = gr.Button(value="Run")
 
             with gr.Column():
-                result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery", columns=[2], rows=[2], height='auto')
+                result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery", columns=[2], height='auto')
         ips = [input_image, prompt, negative_prompt, num_steps, controlnet_conditioning_scale, seed]
         
         run_button.click(fn=process, inputs=ips, outputs=[result_gallery])