Update app.py
Browse files
app.py
CHANGED
|
@@ -83,8 +83,8 @@ def inference(input_img, num_gradcam_images=1, target_layer_number=-1, transpare
|
|
| 83 |
else:
|
| 84 |
return top_n_confidences, visualization, None
|
| 85 |
|
| 86 |
-
title = "CIFAR10 trained on ResNet18 Model with GradCAM"
|
| 87 |
-
description = "A simple Gradio interface to infer on
|
| 88 |
examples = [["cat.jpg", 1, -1, 0.8, True, 3, 3],
|
| 89 |
["dog.jpg", 1, -1, 0.8, True, 3, 3],
|
| 90 |
["plane.jpg", 1, -1, 0.8, True, 3, 3],
|
|
|
|
| 83 |
else:
|
| 84 |
return top_n_confidences, visualization, None
|
| 85 |
|
| 86 |
+
title = "CIFAR10 trained on ResNet18 Model using Pytorch Lightning with GradCAM"
|
| 87 |
+
description = "A simple Gradio interface to infer on ResNet18 model using Pytorch Lightning, and get GradCAM results"
|
| 88 |
examples = [["cat.jpg", 1, -1, 0.8, True, 3, 3],
|
| 89 |
["dog.jpg", 1, -1, 0.8, True, 3, 3],
|
| 90 |
["plane.jpg", 1, -1, 0.8, True, 3, 3],
|