Update app.py

app.py

@@ -22,7 +22,7 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
 #    logging.info(f"RAM memory % used: {psutil.virtual_memory()[2]}")
 
 
-#@st.cache(allow_output_mutation=True, suppress_st_warning=True, max_entries=1)
+@st.cache(allow_output_mutation=True, suppress_st_warning=True, max_entries=1)
 def load_model(model_name):
     return (
         AutoModelForSequenceClassification.from_pretrained(model_name),
@@ -104,35 +104,6 @@ def main():
     else:
         st.write('Goodbye')
     print ("After test button")
-    if st.button("Interpret Text"):
-        #print_memory_usage()
-        st.text("Output")
-        with st.spinner("Interpreting your text (This may take some time)"):
-            print ("Interpreting text")
-            if explanation_class_choice != "predicted":
-                word_attributions = cls_explainer(
-                    text,
-                    class_name=explanation_class_choice,
-                    embedding_type=emb_type_num,
-                    internal_batch_size=2,
-                )
-            else:
-                word_attributions = cls_explainer(
-                    text, embedding_type=emb_type_num, internal_batch_size=2
-                )
-
-        if word_attributions:
-            print ("Word Attributions")
-            word_attributions_expander = st.beta_expander(
-                "Click here for raw word attributions"
-            )
-            with word_attributions_expander:
-                st.json(word_attributions)
-            #components.v1.html(
-            #    cls_explainer.visualize()._repr_html_(), scrolling=True, height=350
-            #)
-        print ("end of stuff")
-
 
 if __name__ == "__main__":
     main()