Merge pull request #6 from joulammous/milestone-2

app.py

@@ -7,18 +7,25 @@ from transformers import pipeline
 
 # Milestone 2
 
+# select a pretrained model
 sentiment_pipeline = pipeline("sentiment-analysis", model = "siebert/sentiment-roberta-large-english")
 
-st.title('Sentiment Analysis with Hugging Face')
-st.write('Enter some text and we will predict its sentiment!')
+# intro
+st.title("Sentiment Analysis App")
+st.write("Enter some text and I'll predict its sentiment!")
 
-# Add a text input box for the user to enter text
-text_input = st.text_input('Enter text here')
+# add a text input box
+text_input = st.text_input("Enter your text here:", value = "The weather is nice today.")
 
-# When the user submits text, run the sentiment analysis model on it
-if st.button('Submit'):
-    # Predict the sentiment of the text using the Hugging Face model
-    sentiment = sentiment_pipeline(text_input)[0]['label']
+# run the model when the user clicks submit
+if st.button("Submit"):
     
-    # Display the sentiment prediction to the user
-    st.write(f'Sentiment: {sentiment}')
+    # get result
+    sentiment = sentiment_pipeline(text_input)
+    
+    # split into sentiment and score
+    sen = sentiment[0]['label']
+    score = round(sentiment[0]['score'], 4)
+    
+    # display the prediction
+    st.write(f"Sentiment: {sen}   ,   Confidence Score: {score}")