Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from pytorch_grad_cam.utils.image import show_cam_on_image
|
|
8 |
from custom_resnet import Assignment12Resnet
|
9 |
import random
|
10 |
import os
|
11 |
-
pl_model = Assignment12Resnet
|
12 |
|
13 |
inv_normalize = transforms.Normalize(
|
14 |
mean=[-0.50/0.23, -0.50/0.23, -0.50/0.23],
|
|
|
8 |
from custom_resnet import Assignment12Resnet
|
9 |
import random
|
10 |
import os
|
11 |
+
pl_model = Assignment12Resnet.load_from_checkpoint("epoch=22-step=4140.ckpt",map_location=torch.device("cpu"))
|
12 |
|
13 |
inv_normalize = transforms.Normalize(
|
14 |
mean=[-0.50/0.23, -0.50/0.23, -0.50/0.23],
|