File size: 988 Bytes
0e46a6f
 
 
 
ee41f8a
0e46a6f
ee41f8a
 
b241691
ee41f8a
 
e25cffd
0e46a6f
 
 
 
 
 
 
4e4d22d
f154ee6
0e46a6f
 
 
ee41f8a
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import streamlit as st
from transformers import pipeline


def run_model(text_in, model_in):
    classifier = pipeline(task="sentiment-analysis",
                          model=model_in)
    analysis = classifier(text_in)
    for output in analysis:
        to_output = "Sentiment: ", output["label"], " Confidence Score: ", "{0:.2g}".format(
            output["score"] * 100)
        st.markdown(to_output)


models_available = {"Roberta Large English": "siebert/sentiment-roberta-large-english",
                    "Generic": "Seethal/sentiment_analysis_generic_dataset",
                    "Twitter Roberta": "cardiffnlp/twitter-roberta-base-sentiment"}

st.title("Sentiment Analysis Web Application")
text_input = st.text_area(
    label="Enter the text to analyze", value="I Love Pizza")
model_picked = st.selectbox(
    "Choose a model to run on", options=models_available.keys())

st.button("Submit", on_click=run_model, args=(
    text_input, models_available[model_picked]))