import streamlit as st
from transformers import pipeline
classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")

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 != "":
    candidate_labels = cats.split(",")
    res = classifier(text, candidate_labels)

    for index, name in enumerate(res['labels']):
        st.text(f"{name} : {round(res['scores'][index], 3) * 100}%")