| import streamlit as st | |
| from transformers import pipeline | |
| def main(): | |
| sentiment_pipeline = pipeline("sentiment-analysis", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english") | |
| st.title("ISOM5240: Sentiment Analysis") | |
| 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.lower()}") | |
| st.write(f"Confidence: {confidence:.4f}") | |
| if __name__ == "__main__": | |
| main() | |