Update app.py

app.py

@@ -52,11 +52,11 @@ def image_classifier(input_image, top_classes=3, show_cam=True, target_layers=[2
     o = softmax(output.flatten())
 
     confidences = {classes[i]: float(o[i]) for i in range(10)}
-    #_, prediction = torch.max(outputs, 1)
+    _, label = torch.max(output, 1)
     
     #confidences.sort(key=lambda x: x[1], reverse=True)
     #confidences = OrderedDict(confidences[:top_classes])
-    label = torch.argmax(output).item()
+    #label = torch.argmax(output).item()
 
     outputs = list()
     if show_cam: