Merge pull request #8 from joulammous/milestone-2

app.py

@@ -7,25 +7,31 @@ from transformers import pipeline
 
 # Milestone 2
 
-# select a pretrained model
-sentiment_pipeline = pipeline("sentiment-analysis", model = "siebert/sentiment-roberta-large-english")
-
-# intro
+# title
 st.title("Sentiment Analysis App")
 st.write("Enter some text and I'll predict its sentiment!")
 
 # add a text input box
 text_input = st.text_input("Enter your text here:", value = "The weather is nice today.")
 
+# use a select box for pretrained models
+option = st.selectbox(
+    "Select a pretrained model for sentiment analysis",
+    ("siebert/sentiment-roberta-large-english", "yiyanghkust/finbert-tone", "finiteautomata/bertweet-base-sentiment-analysis"))
+
+# display selection
+st.write("You selected:", option)
+
+# run pipeline
+sentiment_pipeline = pipeline("sentiment-analysis", model = option)
+
 # run the model when the user clicks submit
 if st.button("Submit"):
-    
-    # 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
+    # get results
     st.write(f"Sentiment: {sen}   ,   Confidence Score: {score}")