Xylor commited on
Commit
5fe3cd7
·
verified ·
1 Parent(s): f8141d5

Change output to somewhat human readable

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -18,7 +18,11 @@ categories = [
18
  ]
19
 
20
  def clf_text(txt: str):
21
- return classifier(txt, categories)
 
 
 
 
22
  # classifier(sequence_to_classify, candidate_labels)
23
  #{'labels': ['travel', 'dancing', 'cooking'],
24
  # 'scores': [0.9938651323318481, 0.0032737774308770895, 0.002861034357920289],
 
18
  ]
19
 
20
  def clf_text(txt: str):
21
+ res = classifier(txt, categories)
22
+ items = sorted(zip(res["labels"], res["scores"]), key=lambda tpl: tpl[1])
23
+ # d = dict(zip(res["labels"], res["scores"]))
24
+ output = [f"{lbl}:\t{score}" for lbl, score in items]
25
+ return "\n".join(output)
26
  # classifier(sequence_to_classify, candidate_labels)
27
  #{'labels': ['travel', 'dancing', 'cooking'],
28
  # 'scores': [0.9938651323318481, 0.0032737774308770895, 0.002861034357920289],