Update app.py

app.py

@@ -31,14 +31,20 @@ def get_score(model_base, text):
 
 # Ask user for input, return scores
 st.title("Toxic Comment Classifier")
-text_input = st.text_input("Enter text for toxicity classification",
-                           "I hope you die")
-submit_btn = st.button("Submit")
+st.write("John Makely")
+st.write("The following model's are fine tuned on the jigsaw-toxic-comment-classification dataset")
+st.write("Please be patient and give the queries and tables time to load (max 2 minutes)")
 
 # Drop down menu for model selection, default is roberta
 model_base = st.selectbox("Select a pretrained model",
                           ["roberta-base", "bert-base-cased", "distilbert-base-cased"])
 
+text_input = st.text_input("Enter text for toxicity classification",
+                           "I hope you die")
+st.write("After hitting Submit, classification scores will be displayed for the provided text")
+submit_btn = st.button("Submit")
+
+
 if submit_btn and text_input:
     result = get_score(model_base, text_input)
 
@@ -70,9 +76,12 @@ scores_df = scores_df.round(2)
 
 
 st.subheader("Toxicity Scores for Random Comments")
-st.table(scores_df)
-
+st.write("The following table will grab random values from the jigsaw dataset and display their respective scores")
+st.write("Please be patient as it may take some time for the scores to be passed through the model")
 # Create a button to reset the scores
 if st.button("Refresh Random Tweets"):
     reset_scores()
     st.success("New tweets have been loaded!")
+st.table(scores_df)
+
+