Spaces:
Sleeping
Sleeping
Update app.py
Browse filesFixed the network architecture name
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
|
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',
|