.

app.py

@@ -6,12 +6,14 @@ from transformers import pipeline
 classifier = pipeline("zero-shot-classification",
                       model="valhalla/distilbart-mnli-12-1")
 
+with st.form('inputs') as form:
+    input_text = form.text_area("Enter your sample")
+    input_label = form.text_input("Enter labels", placeholder="support, help, important")
+    submit_button = form.form_submit_button(label='Submit')
 
-input_text = st.text_area("Enter your sample")
-input_label = st.text_input("Enter labels", placeholder="support, help, important")
+if submit_button:
 
-labels = input_label.split(',')
+    labels = list(l.strip() for l in input_label.split(','))
+    pred = classifier(input_text, labels, multi_class=True)
 
-pred = classifier(input_text, labels, multi_class=True)
-
-st.markdown(pred)
+    st.markdown(pred)