Spaces:
Runtime error
Runtime error
Update utils.py
Browse files
utils.py
CHANGED
@@ -46,5 +46,5 @@ def get_tags_for_prompts(w2v_model, mubert_tags_embeddings, prompts, top_n=3, de
|
|
46 |
top_prob = 1 - scores[idxs[:top_n]]
|
47 |
if debug:
|
48 |
print(f"Prompt: {prompts[i]}\nTags: {', '.join(top_tags)}\nScores: {top_prob}\n\n\n")
|
49 |
-
ret.append((prompts[i], list(top_tags)))
|
50 |
return ret
|
|
|
46 |
top_prob = 1 - scores[idxs[:top_n]]
|
47 |
if debug:
|
48 |
print(f"Prompt: {prompts[i]}\nTags: {', '.join(top_tags)}\nScores: {top_prob}\n\n\n")
|
49 |
+
ret.append((prompts[i], list(top_tags)))
|
50 |
return ret
|