Vvaann commited on
Commit
7e77efb
·
verified ·
1 Parent(s): 4925060

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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", 0, False, 4, 0.8]]
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__":