sentiment-analysis / sentiment-analyser.py
sm2899's picture
Update sentiment-analyser.py
03c4a3b
raw
history blame
No virus
1.15 kB
import streamlit as st
from transformers import pipeline, AutoTokenizer
st.title('Sentiment Analyser App')
st.write('Welcome to my sentiment analysis app!')
model_options=["sentiment-analysis", "twitter-xlm-roberta-base-sentiment"]
form = st.form(key='sentiment-form')
model_type = form.selectbox(label="Select a model", options=model_options)
user_input = form.text_area(label='Enter your text to analyse', value="Hey how are you?")
submit = form.form_submit_button('Submit')
def classification(user_input, type):
if type=="sentiment-analysis":
classifier = pipeline("sentiment-analysis")
elif type=="twitter-xlm-roberta-base-sentiment":
path="cardiffnlp/twitter-xlm-roberta-base-sentiment"
classifier = pipeline("sentiment-analysis", model=path, tokenizer=path)
result = classifier(user_input)[0]
return result
if submit:
resultf = classification(user_input, model_type)
label = resultf['label']
score = resultf['score']
if label == 'POSITIVE' or 'Positive':
st.success(f'{label} sentiment (score: {score})')
else:
st.error(f'{label} sentiment (score: {score})')