moshel commited on
Commit
af18fab
·
1 Parent(s): f4ae5e5
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -5,6 +5,8 @@ import torch
5
  import torchvision
6
  import timm
7
  import cv2
 
 
8
 
9
  print(timm.__version__)
10
 
 
5
  import torchvision
6
  import timm
7
  import cv2
8
+ from pytorch_grad_cam import GradCAM
9
+ from pytorch_grad_cam.utils.image import show_cam_on_image
10
 
11
  print(timm.__version__)
12