Update app.py

app.py

@@ -16,6 +16,6 @@ if submit_button:
     labels = list(l.strip() for l in input_label.split(','))
     pred = classifier(input_text, labels, multi_class=True)
 
-    st.success(pred['label'])
+    st.success(pred['labels'][0:1])
 
     # st.markdown(pred)