Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -94,14 +94,13 @@ demo = gr.Interface(
|
|
| 94 |
gr.Slider(0, 1, value=0.5,label="Overall opacity of the overlay"),
|
| 95 |
],
|
| 96 |
outputs = [
|
| 97 |
-
gr.Label(label="Class", container=True, show_label= True),
|
| 98 |
gr.Image(width= 256, height=256,label="Output Image"),
|
| 99 |
gr.Label(label="Confidences", container=True, show_label= True),
|
| 100 |
],
|
| 101 |
title = "CIFAR 10 trained on ResNet model in pytorch lightning with Gradcam",
|
| 102 |
description = " A simple gradio inference to infer on resnet18 model",
|
| 103 |
examples = [["cat.jpg", 1, True, 10, 0.4],
|
| 104 |
-
["dog.jpg",
|
| 105 |
)
|
| 106 |
|
| 107 |
if __name__ == "__main__":
|
|
|
|
| 94 |
gr.Slider(0, 1, value=0.5,label="Overall opacity of the overlay"),
|
| 95 |
],
|
| 96 |
outputs = [
|
|
|
|
| 97 |
gr.Image(width= 256, height=256,label="Output Image"),
|
| 98 |
gr.Label(label="Confidences", container=True, show_label= True),
|
| 99 |
],
|
| 100 |
title = "CIFAR 10 trained on ResNet model in pytorch lightning with Gradcam",
|
| 101 |
description = " A simple gradio inference to infer on resnet18 model",
|
| 102 |
examples = [["cat.jpg", 1, True, 10, 0.4],
|
| 103 |
+
["dog.jpg", 1, False, 4, 0.8]]
|
| 104 |
)
|
| 105 |
|
| 106 |
if __name__ == "__main__":
|