Update README.md
Browse files
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
|
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)
|