Yong-Sik's picture
add req, streamlit app
c8daa54
raw
history blame contribute delete
No virus
362 Bytes
import streamlit as
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"])