Spaces:
Sleeping
Sleeping
Commit
·
3c39c38
1
Parent(s):
1ffed57
Updated default values
Browse files
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 = [
|