nkanungo commited on
Commit
36de997
·
1 Parent(s): 33e803f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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().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],
 
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],