wipz

app.py

@@ -8,7 +8,7 @@ import numpy as np
 import argparse
 
 from point_e.diffusion.configs import DIFFUSION_CONFIGS, diffusion_from_config
-from .sampler import PointCloudSampler
+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
 
@@ -117,7 +117,7 @@ def generate_3D(input, model_name='base1B', guidance_scale=3.0, grid_size=128):
     set_state('Converting to mesh...')
 
     uniqid = uuid.uuid4()
-    file_path = f'/tmp/mesh-{uniqid}.npz'
+    file_path = f'/tmp/mesh-{uniqid}.npy'
     save_ply(pc, file_path)
 
     set_state('')
@@ -148,10 +148,10 @@ def ply_to_glb(ply_file, glb_file):
 
     return glb_file
 
-def save_ply(pc, file_name):
+def save_plz(pc, file_name):
     # Produce a mesh (with vertex colors)
     with open(file_name, 'wb') as f:
-        pc.save(f)
+        np.save(f, pc.coords)
 
 
 def create_gif(pc):