Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
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)
|