Commit
·
b6cb4be
1
Parent(s):
bb4be9a
Update app.py
Browse files
app.py
CHANGED
|
@@ -142,7 +142,7 @@ class ECCVGenerator(BaseColor):
|
|
| 142 |
|
| 143 |
|
| 144 |
# model_net = torch.load(f'weights/colorizer.pt')
|
| 145 |
-
|
| 146 |
model_net.load_state_dict(torch.load(f'weights/colorizer.pt'))
|
| 147 |
|
| 148 |
|
|
|
|
| 142 |
|
| 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.pt'))
|
| 147 |
|
| 148 |
|