Spaces:
Runtime error
Runtime error
Delete app.py
Browse files
app.py
DELETED
@@ -1,13 +0,0 @@
|
|
1 |
-
import torch,torchvision
|
2 |
-
from torchvision import transforms
|
3 |
-
import numpy as np
|
4 |
-
import gradio as gr
|
5 |
-
from Pytorch_grad_cam import GradCam
|
6 |
-
from Pytorch_grad_cam.utils.image import show_cam_on_image
|
7 |
-
from CUSTOMRESNET import Net
|
8 |
-
|
9 |
-
model=Net()
|
10 |
-
model.load_state_dict(torch.load("model.pth",map_location=torch.device('cpu')),strict=False)
|
11 |
-
|
12 |
-
classes = ('plane', 'car', 'bird', 'cat',
|
13 |
-
'deer', 'dog', 'frog', 'horse', 'ship', 'truck')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|