Update app.py

app.py

@@ -216,7 +216,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 = {name:score for name, score in zip(class_names, result)}
+    labeled_result = {class_names[i]: score for i, score in enumerate(result)}
 
     return labeled_result