Update app.py

app.py

@@ -56,8 +56,8 @@ def image_classifier(input_image, top_classes=3, show_cam=True, target_layers=[2
     confidences = {classes[i]: float(o[i]) for i in range(10)}
     _, label = torch.max(output, 1)
     
-    #confidences.sort(key=lambda x: x[1], reverse=True)
-    #confidences = OrderedDict(confidences[:top_classes])
+    confidences.sort(key=lambda x: x[1], reverse=True)
+    confidences = OrderedDict(confidences[:top_classes])
     #label = torch.argmax(output).item()
 
     outputs = list()
@@ -69,7 +69,7 @@ def image_classifier(input_image, top_classes=3, show_cam=True, target_layers=[2
 
     return outputs, confidences
 
-examples = [["examples/cat.jpg", 0.5, -1], ["examples/dog.jpg", 0.5, -1]]
+examples = [["examples/cat.jpg", 3, True,["-2","-1"],0.5], ["examples/dog.jpg", 3, True,["-2","-1"],0.5]]
 demo1 = gr.Interface(
     fn=image_classifier,
     inputs=[
@@ -82,7 +82,7 @@ demo1 = gr.Interface(
         gr.Slider(0, 1, value=0.5, label="Transparency", step=0.1,
                   info="Set Transparency of CAMs")
     ],
-    outputs=[gr.Gallery(label="Output Images", columns=2, rows=2), gr.Label(label='Top Classes',num_top_classes=3)],
+    outputs=[gr.Gallery(label="Output Images", columns=2, rows=2), gr.Label(label='Top Classes')],
     examples=examples
 )