vibs08 commited on
Commit
311f1b1
·
verified ·
1 Parent(s): 7e664ff

Update app.py

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