Update app.py

app.py

@@ -4,14 +4,12 @@ from transformers import pipeline
 st.title("Text Classifier - BART-Large-MNLI")
 st.write("Input text and labels. Submit. The machine will classify the text according to the labels.")
 
-classifier = pipeline("zero-shot-classification", model='facebook/bart-large-mnli')
+classifier = pipeline(task="zero-shot-classification", device=0, model="facebook/bart-large-mnli")
 
-form = st.form(key='zeroclassifier_form')
-input_text = form.text_input(label='Input Text')
-labels = form.text_input(label='Labels')
-submit_button = form.form_submit_button(label='Submit')
+input_text = st.text_input(label='Input Text')
+labels = st.text_input(label='Labels')
 
-if submit_button:
+if st.button("Submit"):
    labs = labels.title().split(',')
    text = input_text.title() 
    res = classifier(text, labs)