Update app.py

app.py

@@ -1,5 +1,12 @@
+import streamlit as st
 from transformers import pipeline
 classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
 
+st.title("What's the category?")
+
+text = st.text_input("Enter words")
+
 categories = ['lifestyle', 'technology', 'entertainment', 'news']
-classifier(text, categories, multi_class=True)
+res = classifier(text, categories)
+
+st.markdown(f'Well here you go: {res}')