Spaces:
Sleeping
Sleeping
import streamlit as st | |
from transformers import pipeline | |
def main(): | |
sentiment_pipeline = pipeline(model="distilbert/distilbert-base-uncased-finetuned-sst-2-english") | |
st.title("Sentiment Analysis with HuggingFace Spaces") | |
st.write("Enter a sentence to analyze its sentiment:") | |
user_input = st.text_input("") | |
if user_input: | |
result = sentiment_pipeline(user_input) | |
sentiment = result[0]["label"] | |
confidence = result[0]["score"] | |
st.write(f"Sentiment: {sentiment}") | |
st.write(f"Confidence: {confidence:.2f}") | |
if __name__ == "__main__": | |
main() |