Vvaann commited on
Commit
d82f8ec
·
verified ·
1 Parent(s): 74aa401

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ from torchvision import transforms
4
  import numpy as np
5
  import gradio as gr
6
  from PIL import Image
7
- from grad-cam import GradCAM
8
- from grad-cam.utils.image import show_cam_on_image
9
  from resnet_lightning import ResNet18Model
10
  import gradio as gr
11
 
 
4
  import numpy as np
5
  import gradio as gr
6
  from PIL import Image
7
+ from pytorch_grad_cam import GradCAM
8
+ from pytorch_grad_cam.utils.image import show_cam_on_image
9
  from resnet_lightning import ResNet18Model
10
  import gradio as gr
11