Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -48,6 +48,7 @@ def resize_image_pil(image, new_width, new_height):
|
|
| 48 |
|
| 49 |
|
| 50 |
def inference(input_img, show_gradcam, layer_name, num_classes, transparancy = 0.5):
|
|
|
|
| 51 |
input_img = resize_image_pil(input_img,32,32)
|
| 52 |
input_img = np.array(input_img)
|
| 53 |
org_img = input_img
|
|
@@ -101,7 +102,7 @@ demo = gr.Interface(
|
|
| 101 |
],
|
| 102 |
title = "CIFAR 10 trained on ResNet model in pytorch lightning with Gradcam",
|
| 103 |
description = " A simple gradio inference to infer on resnet18 model",
|
| 104 |
-
examples = [["cat.jpg",
|
| 105 |
["dog.jpg", 4, False, "1", -1]]
|
| 106 |
)
|
| 107 |
|
|
|
|
| 48 |
|
| 49 |
|
| 50 |
def inference(input_img, show_gradcam, layer_name, num_classes, transparancy = 0.5):
|
| 51 |
+
print(show_gradcam, layer_name, num_classes, transparancy)
|
| 52 |
input_img = resize_image_pil(input_img,32,32)
|
| 53 |
input_img = np.array(input_img)
|
| 54 |
org_img = input_img
|
|
|
|
| 102 |
],
|
| 103 |
title = "CIFAR 10 trained on ResNet model in pytorch lightning with Gradcam",
|
| 104 |
description = " A simple gradio inference to infer on resnet18 model",
|
| 105 |
+
examples = [["cat.jpg", 10, True, "1", -1],
|
| 106 |
["dog.jpg", 4, False, "1", -1]]
|
| 107 |
)
|
| 108 |
|