lucbelanger's picture
rename streamlit_app.py to app.py
858534d
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"])