sanjanatule commited on
Commit
a342460
·
1 Parent(s): cae6b54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
  from PIL import Image
6
  from pytorch_grad_cam import GradCAM
7
  from pytorch_grad_cam.utils.image import show_cam_on_image
8
- from resnet import ResNet18
9
  import gradio as gr
10
 
11
  class LitResnet(LightningModule):
 
5
  from PIL import Image
6
  from pytorch_grad_cam import GradCAM
7
  from pytorch_grad_cam.utils.image import show_cam_on_image
8
+ from models import custom_resnet
9
  import gradio as gr
10
 
11
  class LitResnet(LightningModule):