matthewwang16czap
add features
925e8f1
raw
history blame
No virus
425 Bytes
import streamlit as st
from transformers import pipeline
import pandas as pd
model_list = ["distilbert-base-uncased-finetuned-sst-2-english", "cardiffnlp/twitter-roberta-base-sentiment", "finiteautomata/bertweet-base-sentiment-analysis"]
sentiment_pipeline = st.selectbox('Which model to predict?', model_list)
text = st.text_input("text for sentiment analysis", value="I love you")
st.write(sentiment_pipeline(text))