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',
|