Update app.py
Browse files
app.py
CHANGED
|
@@ -49,9 +49,11 @@ title = "TSAI S13 Assignment: YOLO V3 trained on PASCAL VOC Dataset"
|
|
| 49 |
description = "A simple Gradio interface to infer on Custom ResNet model, and get GradCAM results. Please use images that belong to any of these classes - 'plane', 'car', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck'."
|
| 50 |
examples = [["cat.jpg","yes", 0.5, -1]
|
| 51 |
]
|
|
|
|
|
|
|
| 52 |
demo = gr.Interface(
|
| 53 |
inference,
|
| 54 |
-
inputs = [gr.Image(shape=(416, 416), label="Input Image")
|
| 55 |
outputs = [gr.Plot(label="Plot")],
|
| 56 |
title = title,
|
| 57 |
description = description,
|
|
|
|
| 49 |
description = "A simple Gradio interface to infer on Custom ResNet model, and get GradCAM results. Please use images that belong to any of these classes - 'plane', 'car', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck'."
|
| 50 |
examples = [["cat.jpg","yes", 0.5, -1]
|
| 51 |
]
|
| 52 |
+
|
| 53 |
+
#gr.Radio(["yes", "no"], label="Show Gradcam"),gr.Slider(0, 1, value = 0.5, label="If yes, Opacity of GradCAM"), gr.Slider(-2, -1, value = -2, step=1, label="If yes, Which Layer?")
|
| 54 |
demo = gr.Interface(
|
| 55 |
inference,
|
| 56 |
+
inputs = [gr.Image(shape=(416, 416), label="Input Image")],
|
| 57 |
outputs = [gr.Plot(label="Plot")],
|
| 58 |
title = title,
|
| 59 |
description = description,
|