Update app.py

app.py

@@ -22,13 +22,19 @@ def classify(text):
     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 return_labels
+    st.text(scores)
+
+    st.text(len(scores))
+    st.text(len(labels))
+    
+    # for i in range(len(scores)):
+    #     if scores[i] > 0.99:
+    #         return_labels.append(labels[i])
+    #     else:
+    #         break
+    # # output = output[0:10]
+    # return return_labels
+    return labels[0:20]
 
 
 text = st.text_input('Enter some text:')  # Input field for new text