app.py

from transformers  import pipeline
import streamlit as st
#from functions import classify

st.write("welcome")
classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
#res = classify("i am a good boy")
#st.write(res)


def main():
    text = st.text_input("Enter the text to classify: ") #input("Enter the text to classify: ")
    labels = ["TOXIC", "AGGRESSIVE", "HATE SPEECH", "INSULTIVE", "SPAM", "NSFW" "NEUTRAL"]
    if text:
        results = classifier(text, labels)
        #st.write(results['sequence'])
        st.write(results['labels'])
        st.write(results['scores'])
    else:
        st.write("Please enter some text to classify")
    
main()