Solomon-y commited on
Commit
79b9949
1 Parent(s): 8073a49

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
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