Spaces:
Running
Running
anhth commited on
Commit ·
02d8fbb
1
Parent(s): 8314c30
Update Model CFG
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ def colorize_image(gray_image: Image.Image):
|
|
| 40 |
return output_image
|
| 41 |
|
| 42 |
# Initialize the model
|
| 43 |
-
model = ColorComicNet(MODEL_CFG)
|
| 44 |
model.load_state_dict(torch.load("./weights/colorizer.pth", map_location=torch.device('cpu')))
|
| 45 |
model.fuse()
|
| 46 |
model.eval()
|
|
|
|
| 40 |
return output_image
|
| 41 |
|
| 42 |
# Initialize the model
|
| 43 |
+
model = ColorComicNet(**MODEL_CFG)
|
| 44 |
model.load_state_dict(torch.load("./weights/colorizer.pth", map_location=torch.device('cpu')))
|
| 45 |
model.fuse()
|
| 46 |
model.eval()
|