Alesco1973's picture
Upload app.py
3764047
raw
history blame contribute delete
365 Bytes
import streamlit as st
from transformers import pipeline
st.title('Hugging Face Demo')
text = st.text_input('Enter text to analyze')
@st.cache_resource
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'])