HemaAM commited on
Commit
a057223
·
1 Parent(s): 690c969

Update app.py

Browse files

Fixed the network architecture name

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ from pytorch_grad_cam import GradCAM
14
  from pytorch_grad_cam.utils.image import show_cam_on_image
15
  from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
16
 
17
- from custom_resnet_model.model import custResNet
18
 
19
  classes = {0: 'airplane',
20
  1: 'automobile',
 
14
  from pytorch_grad_cam.utils.image import show_cam_on_image
15
  from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
16
 
17
+ from model import custResNet
18
 
19
  classes = {0: 'airplane',
20
  1: 'automobile',