Update demo

app.py

@@ -420,8 +420,21 @@ def main():
             if not ply_path or not os.path.exists(ply_path):
                 return ""
             return copy_to_session_and_get_url(ply_path, session_id)
+        
+        # Wrapper to return only the outputs we want to display
+        def _generate_and_filter_outputs(image_path, auto_crop, guidance_scale, random_seed, num_steps):
+            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
 
         gr.Markdown("## FaceLift: Single Image 3D Face Reconstruction\nTurn a single portrait image into a 3D head model and preview it interactively.")
+        
+        gr.Markdown("""
+        ### 💡 Tips for Best Results
+        - **Works best with near-frontal portrait images**
+        - **Note:** These checkpoints were not trained with accessories (glasses, hats, etc.). Portraits containing accessories may produce suboptimal results.
+        """)
+        
         with gr.Row():
             with gr.Column(scale=1):
                 in_image  = gr.Image(type="filepath", label="Input Portrait Image")
@@ -440,26 +453,23 @@ def main():
                     )
 
             with gr.Column(scale=1):
-                out_proc = gr.Image(label="Processed Input")
-                out_multi = gr.Image(label="Multi-view Generation")
+                gr.Markdown("### Interactive Gaussian Splat Viewer")
+                viewer = gr.HTML("<div id='splat-container' style='width:100%;height:600px;background:#000;border:1px solid #ccc;border-radius:8px;'></div>")
+                url_box = gr.Textbox(label="Scene URL (auto-filled)", interactive=False, visible=False)
+                reload_btn = gr.Button("Reload Viewer", size="sm")
+                
                 out_recon = gr.Image(label="3D Reconstruction")
                 out_video = gr.PlayableVideo(label="Turntable Animation")
-                out_ply   = gr.File(label="3D Model (.ply)")
-
-        gr.Markdown("### Interactive Gaussian Splat Viewer")
-        with gr.Row():
-            url_box = gr.Textbox(label="Scene URL (auto-filled)", interactive=False)
-        viewer = gr.HTML("<div id='splat-container' style='width:100%;height:640px'></div>")
-        reload_btn = gr.Button("Reload Viewer")
+                out_ply   = gr.File(label="3D Gaussians (.ply)")
 
         # Initialize per-browser session
         demo.load(fn=_init_session, inputs=None, outputs=session)
 
         # Chain: run → show outputs → prepare viewer URL → load viewer (JS)
         run_btn.click(
-            fn=pipeline.generate_3d_head,
+            fn=_generate_and_filter_outputs,
             inputs=[in_image, auto_crop, guidance, seed, steps],
-            outputs=[out_proc, out_multi, out_recon, out_video, out_ply],
+            outputs=[out_recon, out_video, out_ply],
         ).then(
             fn=_prep_viewer_url,
             inputs=[out_ply, session],