Spaces:
Paused
Paused
Update app.py
Browse files
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]
|