vibs08 commited on
Commit
d979984
·
verified ·
1 Parent(s): 56285bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -144,7 +144,7 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
144
  image = fill_background(image)
145
  return image
146
 
147
- # @spaces.GPU
148
  def generate(image, mc_resolution, formats=["obj", "glb"]):
149
  scene_codes = model(image, device=device)
150
  mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[0]
 
144
  image = fill_background(image)
145
  return image
146
 
147
+ @spaces.GPU
148
  def generate(image, mc_resolution, formats=["obj", "glb"]):
149
  scene_codes = model(image, device=device)
150
  mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[0]