Sentiment-Analysis / src /streamlit_app.py
Reza110's picture
Update src/streamlit_app.py
e13af06 verified
Raw History Blame Contribute Delete
401 Bytes
import streamlit as st
from transformers import pipeline
@st.cache_resource
def load_model():
return pipeline("sentiment-analysis", model="siebert/sentiment-roberta-large-english")
sentiment_analysis = load_model()
st.title("Sentiment Analysis")
rev = st.text_input("Enter a review to analyze its sentiment.")
if st.button("Analyze") and rev:
st.write(sentiment_analysis(rev)[0]['label'])