app.py

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")

candidate_labels = ['lifestyle', 'technology', 'entertainment', 'news']
res = classifier("Today is a good day to have a good day", candidate_labels)

st.markdown(f'Well here you go: {res}')