Change output to json

clf_text()-output to dict instead of text & gradio_interface outputs to json from text

app.py

@@ -21,8 +21,9 @@ 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]
-    return "\n".join(output)
+    # output = [f"{lbl}:\t{score}" for lbl, score in items]
+    # return "\n".join(output)
+    return dict(items)
 # classifier(sequence_to_classify, candidate_labels)
 #{'labels': ['travel', 'dancing', 'cooking'],
 # 'scores': [0.9938651323318481, 0.0032737774308770895, 0.002861034357920289],
@@ -36,6 +37,6 @@ gradio_interface = gradio.Interface(
   # fn = my_inference_function,
   fn = clf_text,
   inputs = "text",
-  outputs = "text"
+  outputs = "json"
 )
 gradio_interface.launch()