Commit
·
8c65b27
1
Parent(s):
a1171a5
Update app.py
Browse files
app.py
CHANGED
|
@@ -143,7 +143,7 @@ class ECCVGenerator(BaseColor):
|
|
| 143 |
|
| 144 |
# model_net = torch.load(f'weights/colorizer.pt')
|
| 145 |
model_net = ECCVGenerator()
|
| 146 |
-
model_net.load_state_dict(torch.load(f'weights/colorizer (1).pt'))
|
| 147 |
|
| 148 |
|
| 149 |
def resize_original(img: Image.Image):
|
|
|
|
| 143 |
|
| 144 |
# model_net = torch.load(f'weights/colorizer.pt')
|
| 145 |
model_net = ECCVGenerator()
|
| 146 |
+
model_net.load_state_dict(torch.load(f'weights/colorizer (1).pt', map_loaction=torch.device('cpu')))
|
| 147 |
|
| 148 |
|
| 149 |
def resize_original(img: Image.Image):
|