vibs08 commited on
Commit
b0bb202
·
verified ·
1 Parent(s): 7fa3b42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -165,7 +165,7 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
165
  image = fill_background(image)
166
  return image
167
 
168
- # @spaces.GPU
169
  def generate(image, mc_resolution, formats=["obj", "glb"]):
170
  scene_codes = model(image, device=device)
171
  mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[0]
 
165
  image = fill_background(image)
166
  return image
167
 
168
+ @spaces.GPU
169
  def generate(image, mc_resolution, formats=["obj", "glb"]):
170
  scene_codes = model(image, device=device)
171
  mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[0]