Update app.py

app.py

@@ -4,10 +4,12 @@ classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnl
 
 st.title("What's the category?")
 
+
+cats = st.text_input("Enter categories (comma separated)")
 text = st.text_input("Enter words")
 
-if text is not None and text is not "":
-    candidate_labels = ['lifestyle', 'technology', 'entertainment', 'news']
+if text is not None and text != "":
+    candidate_labels = cats.split(",")
     res = classifier(text, candidate_labels)
 
     for index, name in enumerate(res['labels']):