julien-c HF staff commited on
Commit
0c57d95
1 Parent(s): a476390

does not work for avocado

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -14,7 +14,8 @@ def load_mesh(im: str, choose: str):
14
  resource = FileResource(im.name)
15
  model.resources.append(resource)
16
  # let's do surgery
17
- model.model.images[0].uri = im.name
 
18
  with tempfile.NamedTemporaryFile(suffix=".glb", delete=False) as file:
19
  model.export(file.name)
20
  return file.name
 
14
  resource = FileResource(im.name)
15
  model.resources.append(resource)
16
  # let's do surgery
17
+ if choose != "avocado":
18
+ model.model.images[0].uri = im.name
19
  with tempfile.NamedTemporaryFile(suffix=".glb", delete=False) as file:
20
  model.export(file.name)
21
  return file.name