Bazedgul's picture
Create app.py
4fe4948
raw
history blame contribute delete
746 Bytes
import streamlit as st
from transformers import pipeline
# Load the sentiment analysis model from Hugging Face
classifier = pipeline("sentiment-analysis")
def predict_sentiment(text):
result = classifier(text)
return result[0]['label'], result[0]['score']
def main():
st.title("Text Classification App")
# User input
text_input = st.text_area("Enter text:", "Type your text here.")
# Prediction
if st.button("Predict"):
if text_input:
sentiment, confidence = predict_sentiment(text_input)
st.write(f"Sentiment: {sentiment}")
st.write(f"Confidence: {confidence:.4f}")
else:
st.warning("Please enter some text.")
if __name__ == "__main__":
main()