display 3d models

app.py

@@ -155,7 +155,7 @@ def process_image(input_pil, remove_bg):
     shutil.rmtree(temp_dir)  # Cleanup temporary folder
     print(output_video)
     torch.cuda.empty_cache()
-    return output_video[0], output_objects
+    return output_video[0], output_objects[1], output_objects[0]
 
 css="""
 div#col-container{
@@ -197,7 +197,7 @@ def gradio_interface():
                             label="Image input", 
                             type="pil",
                             image_mode="RGBA",
-                            height=240
+                            height=480
                         )
         
                         remove_bg = gr.Checkbox(label="Need to remove BG ?", value=False)
@@ -206,7 +206,9 @@ def gradio_interface():
                     
                     with gr.Column(scale=4):
                         output_video= gr.Video(label="Output Video", elem_id="video-out-elm")
-                        output_objects = gr.Files(label=".OBJ outputs")
+                        with gr.Row():
+                            output_object_mesh = gr.Model3D(label=".OBJ Mesh")
+                            output_object_color = gr.Model3D(label=".OBJ colored")
     
             gr.Examples(
                 examples = examples_folder,
@@ -214,7 +216,7 @@ def gradio_interface():
                 examples_per_page = 11
             )
 
-        submit_button.click(process_image, inputs=[input_image, remove_bg], outputs=[output_video, output_objects])
+        submit_button.click(process_image, inputs=[input_image, remove_bg], outputs=[output_video, output_object_mesh, output_object_color])
 
     return app