Update app.py

app.py

@@ -36,7 +36,7 @@ import os
 import argparse
 from omegaconf import OmegaConf
 from rembg import remove
-from collections import defaultdict
+# from collections import defaultdict
 from engine.pose_estimation.pose_estimator import PoseEstimator
 from LHM.utils.face_detector import VGGHeadDetector
 from LHM.utils.hf_hub import wrap_model_hub
@@ -640,8 +640,8 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
         )
 
         # # export ply model
-        print(dump_model_path)
-        gs_model_list[0].save_ply(dump_model_path)
+        # print(dump_model_path)
+        # gs_model_list[0].save_ply(dump_model_path)
 
         # rendering !!!!
         start_time = time.time()
@@ -728,8 +728,8 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
         )
 
 
-        return dump_image_path, dump_video_path, dump_model_path
-        # return dump_image_path, dump_video_path
+        # return dump_image_path, dump_video_path, dump_model_path
+        return dump_image_path, dump_video_path
         # return rgb, dump_image_path, dump_video_path
 
     # def core_fn_export(image, video_params, working_dir):
@@ -860,28 +860,26 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
 
         # show video && ply model
         with gr.Row():
-            with gr.Column(variant='panel', scale=1):
-                with gr.Tabs(elem_id="openlrm_render_model"):
-                    with gr.TabItem('Rendered 3D Model'):
-                        with gr.Row():
-                            output_model = gr.Model3D(label="Rendered 3D Model")
+            # with gr.Column(variant='panel', scale=1):
+            #     with gr.Tabs(elem_id="openlrm_render_model"):
+            #         with gr.TabItem('Rendered 3D Model'):
+            #             with gr.Row():
+            #                 output_model = gr.Model3D(label="Rendered 3D Model")
 
             with gr.Column(variant='panel', scale=1):
                 with gr.Tabs(elem_id="openlrm_render_video"):
                     with gr.TabItem('Rendered Video'):
                         with gr.Row():
                             output_video = gr.Video(label="Rendered Video", format="mp4", height=480, width=270, autoplay=True)
-
-        with gr.Row():
-            download_btn = gr.Button("Download 3D Model")
-            model_download_link = gr.File(label="3D Model File", interactive=False)
+        # with gr.Row():
+        #     download_btn = gr.Button("Download 3D Model")
+        #     model_download_link = gr.File(label="3D Model File", interactive=False)
         
-        download_btn.click(
-            fn=lambda x: x if x else None,
-            inputs=[output_model],
-            outputs=[model_download_link]
-        )
-
+        # download_btn.click(
+        #     fn=lambda x: x if x else None,
+        #     inputs=[output_model],
+        #     outputs=[model_download_link]
+        # )
         working_dir = gr.State()
         submit.click(
             fn=assert_input_image,
@@ -895,6 +893,7 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
             fn=core_fn,
             inputs=[input_image, video_input, working_dir], # video_params refer to smpl dir
             outputs=[processed_image, output_video, output_model],
+            # outputs=[processed_image, output_video, output_model],
         )
 
         demo.queue(max_size=1)