Spaces:
Runtime error
Runtime error
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)
|