added a checkbox with default false for word importances

app.py

@@ -146,24 +146,30 @@ q2i_map = {
 with st.expander('Expand to explore further'):
     st.write(f'Your comment was rated as a QuAL score of **{results["qual"]["label"]}**, indicating **{qual_map[results["qual"]["label"]]}** quality feedback.')
     
+    do_word_importances = st.checkbox("Calculate word importance. This provides more detail on model reasoning below, but takes much longer to compute.",
+                                      value=False)
+    
     st.markdown('### Level of Detail')
     st.write(f"The model identified a **{q1_map[results['q1']['label']]}** level of detail in your comment.")
-    st.write("Below are words that pointed the model toward (green) or against (red) identifying a high level of detail:")
-    with st.spinner("Calculating word importances, may take a while..."):
-        annotated_text(get_explained_words(st.session_state['comment'], models['q1'], 'LABEL_3', RdYlGn_10))
+    if do_word_importances:
+        st.write("Below are words that pointed the model toward (green) or against (red) identifying a high level of detail:")
+        with st.spinner("Calculating word importances, may take a while..."):
+            annotated_text(get_explained_words(st.session_state['comment'], models['q1'], 'LABEL_3', RdYlGn_10))
 
     st.markdown('### Suggestion for Improvement')
     st.write(f"The model **{q2i_map[results['q2i']['label']]}** predict that you provided a suggestion for improvement in your comment.")
-    st.write(f"Below are words that pointed the model toward (green) or against (red) identifying a suggestion for improvement:")
-    with st.spinner("Calculating word importances, may take a while..."):
-        annotated_text(get_explained_words(st.session_state['comment'], models['q2i'], 'LABEL_0', RdYlGn_10))
+    if do_word_importances:
+        st.write(f"Below are words that pointed the model toward (green) or against (red) identifying a suggestion for improvement:")
+        with st.spinner("Calculating word importances, may take a while..."):
+            annotated_text(get_explained_words(st.session_state['comment'], models['q2i'], 'LABEL_0', RdYlGn_10))
 
     if results['q2i']['label'] == 0:
         st.markdown('### Suggestion Linking')
         st.write(f"The model **{q2i_map[results['q3i']['label']]}** predict that you linked your suggestion")
-        st.write(f"Below are words that pointed the model toward (green) or against (red) identifying a linked suggestion:")
-        with st.spinner("Calculating word importances, may take a while..."):
-            annotated_text(get_explained_words(st.session_state['comment'], models['q3i'], 'LABEL_0', RdYlGn_10))
+        if do_word_importances:
+            st.write(f"Below are words that pointed the model toward (green) or against (red) identifying a linked suggestion:")
+            with st.spinner("Calculating word importances, may take a while..."):
+                annotated_text(get_explained_words(st.session_state['comment'], models['q3i'], 'LABEL_0', RdYlGn_10))
 
 
 # annotated_text(to_disp)