Update app.py

app.py

@@ -307,7 +307,7 @@ class CustomImageDataset(Dataset):
             image = self.transform(image)
         return image
         
-@spaces.GPU
+@spaces.GPU(duration=500)
 def invert(image, mask, pcs=10000, epochs=400, weight_decay = 1e-10, lr=1e-1):
     device = "cuda"
     mean.to(device)
@@ -382,7 +382,7 @@ def invert(image, mask, pcs=10000, epochs=400, weight_decay = 1e-10, lr=1e-1):
     return net
 
 
-@spaces.GPU
+@spaces.GPU(duration=500)
 def run_inversion(net, dict, pcs, epochs, weight_decay,lr):
     init_image = dict["background"].convert("RGB").resize((512, 512))
     mask = dict["layers"][0].convert("RGB").resize((512, 512))