Wootang01's picture
Update app.py
7c09720
import streamlit as st
from transformers import pipeline
import gc
st.header("Sentiment Analyzer - BART-Large-MNLI")
st.write("Input text and press control + enter. The machine will present the likelihood that the text can be categorized as positive, negative and neutral.")
classifier = pipeline("zero-shot-classification", model='facebook/bart-large-mnli')
text = st.text_area("Input text and press control + enter.")
candidate_labels = ['positive', 'negative', 'neutral']
if text:
out = classifier(text, candidate_labels)
st.json(out)
del out
gc.collect()