Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| from transformers import pipeline | |
| # Title of the app | |
| st.title("Sentiment Analysis App") #Sets the title of the Streamlit app to "Sentiment Analysis App". | |
| # Input text from the user | |
| user_input = st.text_area("Enter text to analyze") #Creates a text area where users can input the text they want to analyze. | |
| # Select pretrained model | |
| #distilbert-base-uncased-finetuned-sst-2-english: A DistilBERT model fine-tuned for sentiment analysis on the SST-2 dataset. | |
| #nlptown/bert-base-multilingual-uncased-sentiment: A BERT model trained for sentiment analysis on multiple languages. | |
| model_name = st.selectbox("Select a pretrained model", ["distilbert-base-uncased-finetuned-sst-2-english", "nlptown/bert-base-multilingual-uncased-sentiment"]) #Provides a dropdown menu for users to select a pretrained model provided by the HuggingFace Transformers library | |
| # Initialize the sentiment analysis pipeline | |
| sentiment_analysis = pipeline("sentiment-analysis", model=model_name,device=-1) #Initializes the sentiment analysis pipeline using the selected model. | |
| # Perform sentiment analysis when the button is clicked | |
| if st.button("Analyze"): #reates a button labeled "Analyze". When clicked, it triggers the sentiment analysis. | |
| if user_input: | |
| results = sentiment_analysis(user_input) #If text is entered, the sentiment analysis pipeline processes the text | |
| st.write(results) #results are displayed | |
| else: | |
| st.write("Please enter some text to analyze") #If no text is entered, a message prompting the user to enter some text is displayed | |