demo / app.py
ahmedalrashedi's picture
Update app.py
41c9c41
raw
history blame
No virus
1.08 kB
# import streamlit as st
# from transformers import pipeline
# sentiment_analysis = pipeline("sentiment-analysis")
# text = st.text_input("Enter some text")
# if text:
# result = sentiment_analysis(text)
# st.json(result)
import streamlit as st
from transformers import pipeline, AutoModelForSequenceClassification, AutoTokenizer
# Load sentiment analysis model from Hugging Face
model_name = "distilbert-base-uncased-finetuned-sst-2-english"
model = AutoModelForSequenceClassification.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
sentiment_analyzer = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
# Streamlit UI
st.title("Sentiment Analysis App")
# User input
user_input = st.text_input("Enter a sentence:")
if user_input:
# Perform sentiment analysis
results = sentiment_analyzer(user_input)
# Display sentiment and confidence
sentiment = results[0]['label']
confidence = results[0]['score']
st.write(f"Sentiment: {sentiment}")
st.write(f"Confidence: {confidence:.2f}")