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