gupta1912 commited on
Commit
3e1ae80
·
1 Parent(s): 66b8445

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ import itertools
10
  import matplotlib.pyplot as plt
11
  from utils import LitCIFAR10
12
 
13
- model = LitCIFAR10.load_from_checkpoint("model.ckpt")
14
  model.eval()
15
 
16
  classes = ('plane', 'car', 'bird', 'cat',
@@ -120,7 +120,7 @@ def inference(wants_gradcam, n_gradcam, target_layer_number, transparency, wants
120
 
121
  title = "CIFAR10 trained on Custom ResNet Model with GradCAM"
122
  description = "A simple Gradio interface to infer on ResNet model, and get GradCAM results"
123
- examples = [[None, None, None, None, None, None, 'test_'+str(i)+'.jpg', None] for i in range(10)]
124
 
125
  demo = gr.Interface(inference,
126
  inputs = [gr.Checkbox(False, label='Do you want to see GradCAM outputs?'),
 
10
  import matplotlib.pyplot as plt
11
  from utils import LitCIFAR10
12
 
13
+ model = LitCIFAR10.load_from_checkpoint("model/model.ckpt")
14
  model.eval()
15
 
16
  classes = ('plane', 'car', 'bird', 'cat',
 
120
 
121
  title = "CIFAR10 trained on Custom ResNet Model with GradCAM"
122
  description = "A simple Gradio interface to infer on ResNet model, and get GradCAM results"
123
+ examples = [[None, None, None, None, None, None, 'Images/test_'+str(i)+'.jpg', None] for i in range(10)]
124
 
125
  demo = gr.Interface(inference,
126
  inputs = [gr.Checkbox(False, label='Do you want to see GradCAM outputs?'),