dooraven commited on
Commit
48200ac
1 Parent(s): 5559547
Files changed (2) hide show
  1. app.py +1 -1
  2. pc.py +1 -1
app.py CHANGED
@@ -119,7 +119,7 @@ def generate_3D(input, model_name='base1B', guidance_scale=3.0, grid_size=128):
119
  set_state('Converting to mesh...')
120
 
121
  uniqid = uuid.uuid4()
122
- file_path = f"/tmp/mesh-{uniqid}.npy"
123
 
124
  print('file_path', file_path)
125
 
 
119
  set_state('Converting to mesh...')
120
 
121
  uniqid = uuid.uuid4()
122
+ file_path = f"/tmp/mesh-{uniqid}.npz"
123
 
124
  print('file_path', file_path)
125
 
pc.py CHANGED
@@ -52,7 +52,7 @@ class PointCloud:
52
  with open(f, "wb") as writer:
53
  self.save(writer)
54
  else:
55
- np.save(f, self.coords)
56
 
57
  def write_ply(self, raw_f: BinaryIO):
58
  write_ply(
 
52
  with open(f, "wb") as writer:
53
  self.save(writer)
54
  else:
55
+ np.savez(f, self.coords)
56
 
57
  def write_ply(self, raw_f: BinaryIO):
58
  write_ply(