amildravid4292 commited on
Commit
6751ba5
·
verified ·
1 Parent(s): 63755dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -307,7 +307,7 @@ class CustomImageDataset(Dataset):
307
  image = self.transform(image)
308
  return image
309
 
310
- @spaces.GPU(duration=500)
311
  def invert(image, mask, pcs=10000, epochs=400, weight_decay = 1e-10, lr=1e-1):
312
  device = "cuda"
313
  mean.to(device)
@@ -382,7 +382,7 @@ def invert(image, mask, pcs=10000, epochs=400, weight_decay = 1e-10, lr=1e-1):
382
  return net
383
 
384
 
385
- @spaces.GPU(duration=500)
386
  def run_inversion(net, dict, pcs, epochs, weight_decay,lr):
387
  init_image = dict["background"].convert("RGB").resize((512, 512))
388
  mask = dict["layers"][0].convert("RGB").resize((512, 512))
 
307
  image = self.transform(image)
308
  return image
309
 
310
+ @spaces.GPU(duration=200)
311
  def invert(image, mask, pcs=10000, epochs=400, weight_decay = 1e-10, lr=1e-1):
312
  device = "cuda"
313
  mean.to(device)
 
382
  return net
383
 
384
 
385
+ @spaces.GPU(duration=200)
386
  def run_inversion(net, dict, pcs, epochs, weight_decay,lr):
387
  init_image = dict["background"].convert("RGB").resize((512, 512))
388
  mask = dict["layers"][0].convert("RGB").resize((512, 512))