Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
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)
|