Spaces:

Xylor
/

gradio_test_001

Running

Xylor commited on Mar 31

Commit

9d20c0e

verified ·

1 Parent(s): 5fe3cd7

Using multilabel bart instead

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ categories = [
 ]
 def clf_text(txt: str):
-    res = classifier(txt, categories)
     items = sorted(zip(res["labels"], res["scores"]), key=lambda tpl: tpl[1])
     # d = dict(zip(res["labels"], res["scores"]))
     output = [f"{lbl}:\t{score}" for lbl, score in items]

 ]
 def clf_text(txt: str):
+    res = classifier(txt, categories, multi_label=True)
     items = sorted(zip(res["labels"], res["scores"]), key=lambda tpl: tpl[1])
     # d = dict(zip(res["labels"], res["scores"]))
     output = [f"{lbl}:\t{score}" for lbl, score in items]