File size: 555 Bytes
da16f1b
925e8f1
 
da16f1b
5d4dee1
925e8f1
a6943e2
925e8f1
be54301
5d4dee1
be54301
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import streamlit as st
from transformers import pipeline
import pandas as pd

# load data
model_list = ["distilbert-base-uncased-finetuned-sst-2-english", "cardiffnlp/twitter-roberta-base-sentiment", "finiteautomata/bertweet-base-sentiment-analysis"]
model_name = st.selectbox('Which model to predict?', model_list)
text = st.text_input("text for sentiment analysis", value="I love you")

# run to get result
if st.button('Run'):
    sentiment_pipeline = pipeline(model=model_name)
    result = sentiment_pipeline(text)
    st.write(result)