File size: 683 Bytes
dc087e1 3853221 dc087e1 3853221 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import streamlit as st
from transformers import pipeline
from textblob import TextBlob
model_name= "finiteautomata/bertweet-base-sentiment-analysis"
pipe = pipeline("sentiment-analysis")
st.title("Sentiment Analysis")
st.subheader("Which framework would you like to use for sentiment analysis?")
option = st.selectbox('Choose Framework:',('Transformers', 'TextBlob')) # option is stored in this variable
st.subheader("Enter the text you want to analyze")
text = st.text_input('Enter text:') # text is stored in this variable
if option == 'Transformers':
out = pipe(text)
else:
out = TextBlob(text)
out = out.sentiment
st.write("Sentiment of Text: ")
st.write(out) |