wippers

app.py

@@ -12,7 +12,7 @@ from point_e.diffusion.sampler import PointCloudSampler
 from point_e.models.download import load_checkpoint
 from point_e.models.configs import MODEL_CONFIGS, model_from_config
 from point_e.util.plotting import plot_point_cloud
-from point_e.util.pc_to_mesh import marching_cubes_mesh
+from point_e.util.ply_util import ply_util
 
 from diffusers import StableDiffusionPipeline
 
@@ -145,26 +145,10 @@ def ply_to_glb(ply_file, glb_file):
     return glb_file
 
 def save_ply(pc, file_name, grid_size):
-    set_state('Creating SDF model...')
-    sdf_name = 'sdf'
-    sdf_model = model_from_config(MODEL_CONFIGS[sdf_name], device)
-    sdf_model.eval()
-
-    set_state('Loading SDF model...')
-    sdf_model.load_state_dict(load_checkpoint(sdf_name, device))
-
     # Produce a mesh (with vertex colors)
-    mesh = marching_cubes_mesh(
-        pc=pc,
-        model=sdf_model,
-        batch_size=4096,
-        grid_size=grid_size, # increase to 128 for resolution used in evals
-        progress=True,
-    )
-
-    # Write the mesh to a PLY file to import into some other program.
-    with open(file_name, 'wb') as f:
-        mesh.write_ply(f)
+    with open(file_name, "wb") as f:
+        ply_util.write_ply(f, pc)
+
 
 def create_gif(pc):
     fig = plt.figure(facecolor='black', figsize=(4, 4))
@@ -293,4 +277,4 @@ with block:
     img_button.click(fn=generate_3D, inputs=[input_image, model, scale], outputs=[model_3d, model_gif, file_out])
     text_button.click(fn=generate_3D, inputs=[prompt, model, scale], outputs=[model_3d, model_gif, file_out])
 
-block.launch(show_api=False)
+block.launch(show_api=True, share=True)