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)
|