Create app.py

app.py

@@ -0,0 +1,11 @@
+import torch,torchvision
+from torchvision import transforms
+import numpy as np   
+import gradio as gr
+from Pytorch_grad_cam import GradCam
+from Pytorch_grad_cam.utils.image import show_cam_on_image
+from CUSTOMRESNET import Net
+
+model=Net()
+model.load_state_dict(torch.load("model.pth",map_location=torch.device('cpu')),strict=False)
+