File size: 365 Bytes
a33b209
 
 
 
fb9f686
 
3764047
fb9f686
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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'])