Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| from transformers import pipeline | |
| from transformers import AutoTokenizer, AutoModelForSequenceClassification | |
| from huggingface_hub.inference_api import InferenceApi | |
| import os | |
| models = ["junming-qiu/BertToxicClassifier", "cardiffnlp/twitter-xlm-roberta-base-sentiment", "nlptown/bert-base-multilingual-uncased-sentiment", "Tatyana/rubert-base-cased-sentiment-new"] | |
| st.title('Sentiment Analysis Demo') | |
| with st.form("form"): | |
| selection = st.selectbox('Select Transformer:', models) | |
| text = st.text_input('Enter text: ', "I hate people who walk") | |
| submitted = st.form_submit_button('Submit') | |
| if submitted: | |
| model_name = models[models.index(selection)] | |
| if model_name == "junming-qiu/BertToxicClassifier": | |
| API_TOKEN=os.environ['API-KEY'] | |
| inference = InferenceApi(repo_id=model_name, token=API_TOKEN) | |
| predictions = inference(inputs=text)[0] | |
| predictions = sorted(predictions, key=lambda x: x['score'], reverse=True) | |
| hide_table_row_index = """ | |
| <style> | |
| thead tr th:first-child {display:none} | |
| tbody th {display:none} | |
| </style> | |
| """ | |
| st.markdown(hide_table_row_index, unsafe_allow_html=True) | |
| output = [ | |
| {"tweet": text, | |
| "label": predictions[0]['label'], | |
| "score" : predictions[0]['score'], | |
| "label 2": predictions[1]['label'], | |
| "score 2" : predictions[1]['score']}, | |
| ] | |
| st.table(output) | |
| else: | |
| model = AutoModelForSequenceClassification.from_pretrained(model_name) | |
| tokenizer = AutoTokenizer.from_pretrained(model_name) | |
| classifier = pipeline("text-classification", model=model, tokenizer=tokenizer) | |
| result = classifier(text) | |
| st.write("Label:", result[0]["label"]) | |
| st.write('Score: ', result[0]['score']) | |