Spaces:

Gosula
/

ERA_S12

Sleeping

Gosula commited on Aug 14, 2023

Commit

9c7b15b

•

1 Parent(s): c5d735a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from pytorch_grad_cam.utils.model_targets import ClassifierOutputTarget
 from pytorch_grad_cam.utils.image import show_cam_on_image
 import torch
 from torchvision import datasets, transforms
-#from model import CustomResNet
 import random

 from pytorch_grad_cam.utils.image import show_cam_on_image
 import torch
 from torchvision import datasets, transforms
+from custom_resnet import CustomResNet
 import random