Update app.py

app.py

@@ -217,7 +217,7 @@ def image_classifier(inp):
     # postprocess
     result = torch.nn.functional.softmax(result, dim=1) # apply softmax
     result = result[0].detach().numpy().tolist() # take the first batch
-    labeled_result = {class_names[i]: score for i, score in enumerate(result)}
+    labeled_result = {class_names[key]: score for key, score in result.items()}
 
     return labeled_result