Spaces:

lemonaddie
/

geowizard

Running on Zero

lemonaddie commited on Mar 28

Commit

0b9f1b0

•

1 Parent(s): a34258b

Update app_recon.py

Files changed (1) hide show

app_recon.py CHANGED Viewed

@@ -197,7 +197,7 @@ def seg_foreground(image_file):
     return masked_image, mask
-@spaces.GPU
 def reconstruction(image_file, files):
     torch.cuda.empty_cache()

     return masked_image, mask
+@spaces.GPU(duration=120)
 def reconstruction(image_file, files):
     torch.cuda.empty_cache()