almontalvao's picture
Update app.py
cb8f2b5
import streamlit as st
from transformers import pipeline
#Load model from HF
clf = pipeline('sentiment-analysis')
#Create Streamlit app
st.title('Sentiment Analysis')
st.write('Enter a text and we will predict its sentiment!')
#Add text box
txt_input = st.text_input('Enter text here:')
#When text is entered, run model
if st.button('Submit'):
#Predict the sentiment using HF model
sentiment = clf(txt_input)[0]['label']
#Display prediction
st.write(f'Sentiment: {sentiment}')