vibs08 commited on
Commit
38c07d5
·
verified ·
1 Parent(s): e4ffaa9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -177,7 +177,7 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
177
  torch.cuda.empty_cache()
178
  return image
179
 
180
- # @spaces.GPU
181
  def generate(image, mc_resolution, formats=["obj", "glb"]):
182
  torch.cuda.synchronize()
183
  scene_codes = model(image, device=device)
 
177
  torch.cuda.empty_cache()
178
  return image
179
 
180
+ @spaces.GPU
181
  def generate(image, mc_resolution, formats=["obj", "glb"]):
182
  torch.cuda.synchronize()
183
  scene_codes = model(image, device=device)