AkashDataScience commited on
Commit
3c39c38
·
1 Parent(s): 1ffed57

Updated default values

Browse files
Files changed (1) hide show
  1. app.py +10 -14
app.py CHANGED
@@ -24,10 +24,6 @@ dataloader_args = dict(shuffle=True, batch_size=128, num_workers=4, pin_memory=T
24
 
25
  test_loader = dataset.get_test_data_loader(**dataloader_args)
26
 
27
- inv_normalize = transforms.Normalize(
28
- mean=[-0.48215841/0.24348513, -0.44653091/0.26158784, -0.49139968/0.24703223],
29
- std=[1/0.24348513, 1/0.26158784, 1/0.24703223]
30
- )
31
  classes = ('plane', 'car', 'bird', 'cat', 'deer',
32
  'dog', 'frog', 'horse', 'ship', 'truck')
33
 
@@ -100,16 +96,16 @@ def inference(input_img, is_grad_cam=True, transparency = 0.5, target_layer_numb
100
 
101
  title = "CIFAR10 trained on ResNet18 Model with GradCAM"
102
  description = "A simple Gradio interface to infer on ResNet model, and get GradCAM results"
103
- examples = [["cat.jpg", True, 0.5, -1, 3],
104
- ["dog.jpg", True, 0.5, -1, 3],
105
- ["bird.jpg", True, 0.5, -1, 3],
106
- ["car.jpg", True, 0.5, -1, 3],
107
- ["deer.jpg", True, 0.5, -1, 3],
108
- ["frog.jpg", True, 0.5, -1, 3],
109
- ["horse.jpg", True, 0.5, -1, 3],
110
- ["plane.jpg", True, 0.5, -1, 3],
111
- ["ship.jpg", True, 0.5, -1, 3],
112
- ["truck.jpg", True, 0.5, -1, 3]]
113
  demo = gr.Interface(
114
  inference,
115
  inputs = [
 
24
 
25
  test_loader = dataset.get_test_data_loader(**dataloader_args)
26
 
 
 
 
 
27
  classes = ('plane', 'car', 'bird', 'cat', 'deer',
28
  'dog', 'frog', 'horse', 'ship', 'truck')
29
 
 
96
 
97
  title = "CIFAR10 trained on ResNet18 Model with GradCAM"
98
  description = "A simple Gradio interface to infer on ResNet model, and get GradCAM results"
99
+ examples = [["cat.jpg", True, 0.5, -1, 3, True, 10],
100
+ ["dog.jpg", True, 0.5, -1, 3, True, 10],
101
+ ["bird.jpg", True, 0.5, -1, 3, True, 10],
102
+ ["car.jpg", True, 0.5, -1, 3, True, 10],
103
+ ["deer.jpg", True, 0.5, -1, 3, True, 10],
104
+ ["frog.jpg", True, 0.5, -1, 3, True, 10],
105
+ ["horse.jpg", True, 0.5, -1, 3, True, 10],
106
+ ["plane.jpg", True, 0.5, -1, 3, True, 10],
107
+ ["ship.jpg", True, 0.5, -1, 3, True, 10],
108
+ ["truck.jpg", True, 0.5, -1, 3, True, 10]]
109
  demo = gr.Interface(
110
  inference,
111
  inputs = [