skytnt commited on
Commit
23d2ee2
1 Parent(s): c1d6a4f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -122,7 +122,7 @@ def process(all_images, batch_size=8, score_threshold=0.35):
122
  tags = []
123
  for prob, label in zip(probs[i].tolist(), tagger_tags):
124
  if prob > score_threshold:
125
- tags.append((label.replace("_", " "), prob))
126
  predictions[images[idxs[i].item()]] = tags
127
 
128
  gpu_num = len(tagger_model)
 
122
  tags = []
123
  for prob, label in zip(probs[i].tolist(), tagger_tags):
124
  if prob > score_threshold:
125
+ tags.append((label, prob))
126
  predictions[images[idxs[i].item()]] = tags
127
 
128
  gpu_num = len(tagger_model)