ifire commited on
Commit
06b8314
1 Parent(s): fbf185f

Refactored mesh saving process

Browse files
Files changed (1) hide show
  1. predict.py +7 -10
predict.py CHANGED
@@ -13,15 +13,6 @@ class Predictor(BasePredictor):
13
  "MarcusLoren/MeshGPT-preview"
14
  )
15
 
16
- def save_as_obj(self, file_path):
17
- v, f = igl.read_triangle_mesh(file_path)
18
- v, f, _, _ = igl.remove_unreferenced(v, f)
19
- c, _ = igl.orientable_patches(f)
20
- f, _ = igl.orient_outward(v, f, c)
21
- igl.write_triangle_mesh(file_path, v, f)
22
- output_path = Path(tempfile.mkdtemp()) / file_path
23
- return output_path
24
-
25
  def predict(
26
  self,
27
  text: str = Input(description="Enter labels, separated by commas"),
@@ -62,7 +53,13 @@ class Predictor(BasePredictor):
62
  (self.transformer.generate(texts=labels, temperature=num_temp))
63
  )
64
  file_name = "./mesh.obj"
65
- mesh_render.save_rendering(file_name, output)
 
 
 
 
 
 
66
  file_path = self.save_as_obj(file_name)
67
  return file_path
68
 
 
13
  "MarcusLoren/MeshGPT-preview"
14
  )
15
 
 
 
 
 
 
 
 
 
 
16
  def predict(
17
  self,
18
  text: str = Input(description="Enter labels, separated by commas"),
 
53
  (self.transformer.generate(texts=labels, temperature=num_temp))
54
  )
55
  file_name = "./mesh.obj"
56
+ file_path = Path(tempfile.mkdtemp()) / file_name
57
+ mesh_render.save_rendering(file_path, output)
58
+ v, f = igl.read_triangle_mesh(file_path)
59
+ v, f, _, _ = igl.remove_unreferenced(v, f)
60
+ c, _ = igl.orientable_patches(f)
61
+ f, _ = igl.orient_outward(v, f, c)
62
+ igl.write_triangle_mesh(file_path, v, f)
63
  file_path = self.save_as_obj(file_name)
64
  return file_path
65