Spaces:
Paused
Paused
Update utils.py
Browse files
utils.py
CHANGED
@@ -32,4 +32,5 @@ def get_tags_for_prompts(w2v_model, mubert_tags_embeddings, prompts, top_n=3, de
|
|
32 |
if debug:
|
33 |
print(f"Prompt: {prompts[i]}\nTags: {', '.join(top_tags)}\nScores: {top_prob}\n\n\n")
|
34 |
ret.append((prompts[i], list(top_tags)))
|
|
|
35 |
return ret
|
|
|
32 |
if debug:
|
33 |
print(f"Prompt: {prompts[i]}\nTags: {', '.join(top_tags)}\nScores: {top_prob}\n\n\n")
|
34 |
ret.append((prompts[i], list(top_tags)))
|
35 |
+
print("ret: " + ret)
|
36 |
return ret
|