vibs08 commited on
Commit
ec2b233
·
verified ·
1 Parent(s): 95ec856

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -170,7 +170,7 @@ def preprocess(input_image, do_remove_background, foreground_ratio):
170
  torch.cuda.synchronize()
171
  return image
172
 
173
- @spaces.GPU
174
  def generate(image, mc_resolution, formats=["obj", "glb"]):
175
  scene_codes = model(image, device=device)
176
  mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[0]
 
170
  torch.cuda.synchronize()
171
  return image
172
 
173
+ # @spaces.GPU
174
  def generate(image, mc_resolution, formats=["obj", "glb"]):
175
  scene_codes = model(image, device=device)
176
  mesh = model.extract_mesh(scene_codes, resolution=mc_resolution)[0]