Update app.py

app.py

@@ -19,9 +19,16 @@ with open('chapter_titles.pkl', 'rb') as file:
 
 def classify(text):
     output = classifier(text, titles_astiko, multi_label=True)
-    # output = output['sequence']
+    labels = output['labels']
+    scores = output['scores']
+    return_labels = []
+    for i in range(len(scores)):
+        if scores[i] > 0.99:
+            return_labels.append(labels[i])
+        else:
+            break
     # output = output[0:10]
-    return output
+    return return_labels
 
 
 text = st.text_input('Enter some text:')  # Input field for new text