File size: 1,112 Bytes
92387b3
bcf37b9
 
18ad580
 
e8c2178
92387b3
18ad580
e8c2178
18ad580
 
e8c2178
18ad580
da1ec5b
b236a9b
 
 
 
 
 
 
 
 
 
 
 
 
 
da1ec5b
 
b236a9b
 
 
 
 
da1ec5b
 
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
28
29
30
31
32
33
34
35
36
37
38
import streamlit as st
from transformers import pipeline

# title
st.title('Analyze Sentiment of Text!')

# Display text
st.write("You will get a sentiment analysis after you enter something.")

# show an emoji
st.write("Let's try it: :smiley:")

user_input = st.text_input('Enter something below:')

model_options = ["finiteautomata/bertweet-base-sentiment-analysis", 
                 "nlptown/bert-base-multilingual-uncased-sentiment", 
                 "bhadresh-savani/distilbert-base-uncased-emotion"]
selected_model = st.selectbox('Choose a model:', model_options)

def choose_model(option):
    chosen_model = pipeline("sentiment-analysis", model=option)
    return chosen_model

# Initialize the model
model = None
if selected_model:
    model = choose_model(selected_model)

if st.button('Submit'):
    if user_input:
        if model:
            results = model(user_input)
            st.write('The detected sentiment is', results[0]['label'], "with a score of", results[0]['score'])
        else:
            st.write('Please select a model!')
    else:
        st.write('Please enter something!')