Update app.py

app.py

@@ -29,7 +29,7 @@ def classify(image):
     tops = torch.topk(output, k=k).indices
     scores = torch.softmax(output, dim=0)[tops]
     
-    result = {index_to_species[str(tops[i].item())]: round(scores[i].item(), 2) for i in range(len(tops))}
+    result = {index_to_species[str(tops[i].item())].replace("_", " "): round(scores[i].item(), 2) for i in range(len(tops))}
     sorted_result = {k: v for k, v in sorted(result.items(), key=lambda item: item[1], reverse=True) if v > 0}
     
     return sorted_result