Spaces:
Build error
Build error
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
|