Update README.md

README.md

@@ -122,7 +122,7 @@ def process(all_images, batch_size=8, score_threshold=0.35):
                 tags = []
                 for prob, label in zip(probs[i].tolist(), tagger_tags):
                     if prob > score_threshold:
-                        tags.append((label.replace("_", " "), prob))
+                        tags.append((label, prob))
                 predictions[images[idxs[i].item()]] = tags
 
     gpu_num = len(tagger_model)