demo

app.py

@@ -50,11 +50,6 @@ import sys
 OUTPUTS_DIR = Path.cwd() / "outputs"
 OUTPUTS_DIR.mkdir(exist_ok=True)
 
-def _as_str_path(p):
-    """Coerce any Path-like to a plain string for Gradio components."""
-    from pathlib import Path as _Path
-    return str(p) if isinstance(p, (str, _Path)) else ""
-
 def _log_viewer_file(ply_path: Path):
     """Print a concise JSON line about the viewer file so users can debug from Space logs."""
     info = {
@@ -366,12 +361,12 @@ class FaceLiftPipeline:
             # Final CUDA cache clear
             torch.cuda.empty_cache()
             
-            # Return a Model3D.update for the viewer and a string path for the File component
+            # Return paths directly (Gradio components handle them automatically)
             return (
                 str(output_path),              # Reconstruction grid
                 str(turntable_path),           # Turntable video
-                gr.Model3D.update(value=_as_str_path(ply_path)),  # Viewer
-                _as_str_path(ply_path),        # Download file
+                str(ply_path),                 # Viewer
+                str(ply_path),                 # Download file
             )
             
         except Exception as e:
@@ -428,23 +423,8 @@ def main():
                 out_ply    = gr.File(label="Download 3D Model (.ply)")
 
         # Wrapper: call the pipeline and forward outputs in the exact order expected
-        @spaces.GPU(duration=1)  # trivial; only formats outputs
         def _generate_and_filter_outputs(image_path, auto_crop, guidance_scale, random_seed, num_steps):
-            (
-                output_path,
-                turntable_path,
-                viewer_update,
-                ply_dl_path,
-            ) = pipeline.generate_3d_head(image_path, auto_crop, guidance_scale, random_seed, num_steps)
-            # Ensure correct types for each component
-            if not isinstance(viewer_update, gr.components.Model3D.Update):
-                viewer_update = gr.Model3D.update(value=_as_str_path(viewer_update))
-            return (
-                _as_str_path(output_path),
-                _as_str_path(turntable_path),
-                viewer_update,
-                _as_str_path(ply_dl_path),
-            )
+            return pipeline.generate_3d_head(image_path, auto_crop, guidance_scale, random_seed, num_steps)
 
         # Run generation and display all outputs
         run_btn.click(
@@ -457,7 +437,7 @@ def main():
         def _load_sample_viewer():
             sample_path = _write_sample_cube_ply(Path("outputs") / "sample_cube.ply")
             print("[VIEWER-SAMPLE]", os.path.abspath(sample_path))
-            return gr.Model3D.update(value=_as_str_path(sample_path))
+            return sample_path
 
         sample_btn.click(_load_sample_viewer, inputs=[], outputs=[out_viewer])