Spaces:
Running
Running
Change output to somewhat human readable
Browse files
app.py
CHANGED
@@ -18,7 +18,11 @@ categories = [
|
|
18 |
]
|
19 |
|
20 |
def clf_text(txt: str):
|
21 |
-
|
|
|
|
|
|
|
|
|
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],
|