Change return_all_tensors to top_k=None for classification pipeline

app.py

@@ -122,4 +122,4 @@ if __name__ == "__main__":
     demo = gr.TabbedInterface(
         demo_interface["demo"], demo_interface["titles"], theme="soft"
     )
-    demo.launch(debug=True)
+    demo.launch()

utils.py

@@ -33,7 +33,7 @@ def sentence_similarity(text: str, documents: list[str], pipe: SentenceSimilarit
 # Text Analysis
 def cls_inference(input: list[str], pipe: pipeline) -> str:
     results = pipe(input, top_k=None)
-    return {x["label"]: x["score"] for x in results[0]}
+    return {x["label"]: x["score"] for x in results}
 
 
 def text_interface(