linoyts HF staff commited on
Commit
690d2fa
·
verified ·
1 Parent(s): 82ba5bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -401,7 +401,7 @@ def run_inversion(net, dict, pcs, epochs, weight_decay,lr):
401
  return net, net, image
402
 
403
 
404
- #@spaces.GPU
405
  def file_upload(file, net):
406
  device="cuda"
407
  weights = torch.load(file.name).to(device)
 
401
  return net, net, image
402
 
403
 
404
+ @spaces.GPU
405
  def file_upload(file, net):
406
  device="cuda"
407
  weights = torch.load(file.name).to(device)