app.py

app.py

@@ -639,19 +639,8 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
             },
         )
 
-        # export ply model
-        try:
-            print("Attempting PLY export...")
-            export_smplx = {k: v.to("cuda") for k, v in smplx_params.items()}
-            export_smplx['transform_mat_neutral_pose'] = transform_mat_neutral_pose
-            output_gs = lhm.animation_infer_gs(gs_model_list, query_points, export_smplx)
-            output_gs.save_ply(dump_model_path)
-            print(f"PLY saved to {dump_model_path}")
-        except Exception as e:
-            print(f"PLY export failed: {e}")
-            import traceback
-            traceback.print_exc()
-            dump_model_path = None
+        # PLY export disabled for now - debugging
+        dump_model_path = None
 
         # rendering !!!!
         start_time = time.time()
@@ -739,10 +728,7 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
 
 
 
-        if dump_model_path and os.path.exists(dump_model_path):
-            return dump_image_path, dump_video_path, dump_model_path
-        else:
-            return dump_image_path, dump_video_path, None
+        return dump_image_path, dump_video_path
         # return rgb, dump_image_path, dump_video_path
 
     # def core_fn_export(image, video_params, working_dir):
@@ -873,11 +859,11 @@ 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"):
@@ -897,7 +883,7 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
         ).success(
             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],
         )
 
         demo.queue(max_size=1)