demo

app.py

@@ -351,7 +351,7 @@ def main():
             input_path, multiview_path, output_path, turntable_path, ply_path = \
                 pipeline.generate_3d_head(image_path, auto_crop, guidance_scale, random_seed, num_steps)
             
-            return output_path, turntable_path, ply_path
+            return output_path, turntable_path, str(ply_path), ply_path
 
         gr.Markdown("## FaceLift: Single Image 3D Face Reconstruction.")
         
@@ -382,13 +382,15 @@ def main():
             with gr.Column(scale=1):
                 out_recon = gr.Image(label="3D Reconstruction Views")
                 out_video = gr.PlayableVideo(label="Turntable Animation (360° View)", height=600)
+                # Interactive 3D viewer for the generated Gaussian PLY (uses three.js under the hood)
+                out_viewer = gr.Model3D(label="Interactive 3D Viewer (.ply)", height=600)
                 out_ply   = gr.File(label="Download 3D Model (.ply)")
 
         # Run generation and display all outputs
         run_btn.click(
             fn=_generate_and_filter_outputs,
             inputs=[in_image, auto_crop, guidance, seed, steps],
-            outputs=[out_recon, out_video, out_ply],
+            outputs=[out_recon, out_video, out_viewer, out_ply],
         )
 
         demo.queue(max_size=10)