replaced a set of images with a Gallery

app.py

@@ -232,7 +232,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
 
 
 def process_3d(mode, data_dir, guidance_scale, crop_size):
@@ -391,28 +391,16 @@ def run_demo():
                 # gr.Markdown("<span style='color:red'>First click Generate button, then click Reconstruct button. Reconstruction may cost several minutes.</span>")
         
         with gr.Row():
-            view_1 = gr.Image(interactive=False, height=512, show_label=False)
-            view_2 = gr.Image(interactive=False, height=512, show_label=False)
-            view_3 = gr.Image(interactive=False, height=512, show_label=False)
-        with gr.Row():
-            view_4 = gr.Image(interactive=False, height=512, show_label=False)
-            view_5 = gr.Image(interactive=False, height=512, show_label=False)
-            view_6 = gr.Image(interactive=False, height=512, show_label=False)
-        with gr.Row():
-            normal_1 = gr.Image(interactive=False, height=512, show_label=False)
-            normal_2 = gr.Image(interactive=False, height=512, show_label=False)
-            normal_3 = gr.Image(interactive=False, height=512, show_label=False)
-        with gr.Row():
-            normal_4 = gr.Image(interactive=False, height=512, show_label=False)
-            normal_5 = gr.Image(interactive=False, height=512, show_label=False)
-            normal_6 = gr.Image(interactive=False, height=512, show_label=False)
+            view_gallery = gr.Gallery(label='Multiview Images')
+            normal_gallery = gr.Gallery(label='Multiview Normals')
+            
         print('Launching...')
         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, output_processing],
-            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],
         )
         # recon_btn.click(
         #     process_3d, inputs=[mode, data_dir, scale_slider, crop_size], outputs=[obj_3d]