fffiloni commited on
Commit
a989a97
โ€ข
1 Parent(s): 0aab8e8

Update utils.py

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