jonnyzhong's picture
Create app.py
4bac13a
raw
history blame contribute delete
370 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"])