Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ model = Network(img_channel=opt['network']['img_channels'],
|
|
29 |
dilations=opt['network']['dilations'],
|
30 |
extra_depth_wise = opt['network']['extra_depth_wise'])
|
31 |
|
32 |
-
checkpoints = torch.load(
|
33 |
# print(checkpoints)
|
34 |
model.load_state_dict(checkpoints['model_state_dict'])
|
35 |
|
|
|
29 |
dilations=opt['network']['dilations'],
|
30 |
extra_depth_wise = opt['network']['extra_depth_wise'])
|
31 |
|
32 |
+
checkpoints = torch.load('Network_noFAC_LOLBlur.pt', map_location=device)
|
33 |
# print(checkpoints)
|
34 |
model.load_state_dict(checkpoints['model_state_dict'])
|
35 |
|