Spaces:
Runtime error
Runtime error
Commit ·
bd26b85
1
Parent(s): 50ad538
Update app.py
Browse files
app.py
CHANGED
|
@@ -10,6 +10,7 @@ from model import CustomResNet
|
|
| 10 |
PATH = "model.pth"
|
| 11 |
model= CustomResNet()
|
| 12 |
model.load_state_dict(torch.load("model.pth", map_location=torch.device('cpu')), strict=False)
|
|
|
|
| 13 |
|
| 14 |
inv_normalize = transforms.Normalize(
|
| 15 |
mean=[-0.50/0.23, -0.50/0.23, -0.50/0.23],
|
|
|
|
| 10 |
PATH = "model.pth"
|
| 11 |
model= CustomResNet()
|
| 12 |
model.load_state_dict(torch.load("model.pth", map_location=torch.device('cpu')), strict=False)
|
| 13 |
+
model.setup(stage="test")
|
| 14 |
|
| 15 |
inv_normalize = transforms.Normalize(
|
| 16 |
mean=[-0.50/0.23, -0.50/0.23, -0.50/0.23],
|