Spaces:
Running
on
Zero
Running
on
Zero
Update app_recon.py
Browse files- app_recon.py +1 -1
app_recon.py
CHANGED
|
@@ -197,7 +197,7 @@ def seg_foreground(image_file):
|
|
| 197 |
|
| 198 |
return masked_image, mask
|
| 199 |
|
| 200 |
-
@spaces.GPU
|
| 201 |
def reconstruction(image_file, files):
|
| 202 |
|
| 203 |
torch.cuda.empty_cache()
|
|
|
|
| 197 |
|
| 198 |
return masked_image, mask
|
| 199 |
|
| 200 |
+
@spaces.GPU(duration=120)
|
| 201 |
def reconstruction(image_file, files):
|
| 202 |
|
| 203 |
torch.cuda.empty_cache()
|