import streamlit as st | |
from transformers import pipeline | |
st.title('Hugging Face Demo') | |
text = st.text_input('Enter text to analyze') | |
def get_model(): | |
return pipeline("sentiment-analysis") | |
model = get_model() | |
if text: | |
result = model(text) | |
st.write('Sentiment: ', result[0]['label']) | |
st.write('Confidence: ', result[0]['score']) | |