Update gradio_app.py

gradio_app.py

@@ -215,7 +215,7 @@ def run_pipeline(pipeline, cfg, single_image, guidance_scale, steps, seed, crop_
     images_pred = [save_image(images_pred[i]) for i in range(bsz)]
 
     out = images_pred + normals_pred
-    return out
+    return images_pred, normals_pred
 
 
 @dataclass
@@ -285,7 +285,7 @@ def run_demo():
                 gr.Examples(
                     examples=example_fns,
                     inputs=[input_image],
-                    outputs=[input_image],
+                    #outputs=[input_image],
                     cache_examples=False,
                     label='Examples (click one of the images below to start)',
                     examples_per_page=30
@@ -315,27 +315,15 @@ def run_demo():
                     # crop_size = 192
                 run_btn = gr.Button('Generate', variant='primary', interactive=True)
         with gr.Row():
-            view_1 = gr.Image(interactive=False, height=240, show_label=False)
-            view_2 = gr.Image(interactive=False, height=240, show_label=False)
-            view_3 = gr.Image(interactive=False, height=240, show_label=False)
-            view_4 = gr.Image(interactive=False, height=240, show_label=False)
-            view_5 = gr.Image(interactive=False, height=240, show_label=False)
-            view_6 = gr.Image(interactive=False, height=240, show_label=False)
-        with gr.Row():
-            normal_1 = gr.Image(interactive=False, height=240, show_label=False)
-            normal_2 = gr.Image(interactive=False, height=240, show_label=False)
-            normal_3 = gr.Image(interactive=False, height=240, show_label=False)
-            normal_4 = gr.Image(interactive=False, height=240, show_label=False)
-            normal_5 = gr.Image(interactive=False, height=240, show_label=False)
-            normal_6 = gr.Image(interactive=False, height=240, show_label=False)
-        
+            view_gallery = gr.Gallery(interactive=False,show_label=False, container=True, preview=True,  allow_preview=True, height=400)
+            normal_gallery = gr.Gallery(interactive=False,show_label=False, container=True, preview=True, allow_preview=True, height=400 )
 
         first_stage = run_btn.click(fn=partial(preprocess, predictor), 
                         inputs=[input_image, input_processing], 
                         outputs=[processed_image_highres, processed_image], queue=True
             ).success(fn=partial(run_pipeline, pipeline, cfg), 
                         inputs=[processed_image_highres, scale_slider, steps_slider, seed, crop_size],
-                        outputs=[view_1, view_2, view_3, view_4, view_5, view_6, normal_1, normal_2, normal_3, normal_4, normal_5, normal_6]
+                        outputs=[view_gallery, normal_gallery]
                         )
         
         demo.queue().launch(share=True, max_threads=80)