Upload 2 files

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 images_pred, normals_pred
+    return *out, 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
@@ -297,12 +297,15 @@ def run_demo():
                 with gr.Accordion('Advanced options', open=True):
                     with gr.Row():
                         with gr.Column():
-                            input_processing = gr.CheckboxGroup(['Background Removal'], label='Input Image Preprocessing', value=['Background Removal'])
+                            input_processing = gr.CheckboxGroup(['Background Removal'], 
+                                                                label='Input Image Preprocessing',
+                                                                 value=['Background Removal'],
+                                                                 info='untick this, if masked image with alpha channel')
                         with gr.Column():
                             output_processing = gr.CheckboxGroup(['Background Removal'], label='Output Image Postprocessing', value=[]) 
                     with gr.Row():
                         with gr.Column():
-                            scale_slider = gr.Slider(1, 10, value=3, step=1,
+                            scale_slider = gr.Slider(1, 5, value=3, step=1,
                                                         label='Classifier Free Guidance Scale')
                         with gr.Column():
                             steps_slider = gr.Slider(15, 100, value=50, step=1,
@@ -311,21 +314,38 @@ def run_demo():
                         with gr.Column():
                             seed = gr.Number(42, label='Seed')
                         with gr.Column():
-                            crop_size = gr.Number(192, label='Crop size')
+                            crop_size = gr.Number(210, label='Crop size')
                     # crop_size = 192
                 run_btn = gr.Button('Generate', variant='primary', interactive=True)
         with gr.Row():
-            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 )
+            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)
 
-        first_stage = run_btn.click(fn=partial(preprocess, predictor), 
+        with gr.Row():
+            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)
+        
+
+        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_gallery, normal_gallery]
+                        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, 
+                                 view_gallery, normal_gallery]
                         )
-        
         demo.queue().launch(share=True, max_threads=80)
 
 

requirements.txt

@@ -28,5 +28,4 @@ torch_efficient_distloss
 tensorboard
 rembg
 segment_anything
-streamlit==1.22.0
-fire
+