Add: session state

app.py

@@ -8,8 +8,6 @@ import math
 import re
 import os
 
-print("OK")
-
 def set_seed(seed):
     torch.manual_seed(seed)
     if torch.cuda.is_available():
@@ -18,10 +16,14 @@ set_seed(1212)
 
 def loadEnModel():
     os.system("python3 -m spacy download en_core_web_sm")
+    st.session_state['models_loaded'] = True
 
 class GramformerDemo:
 
     def __init__(self):
+    	if 'models_loaded' not in st.session_state:
+    		st.session_state['models_loaded'] = False
+    		
         st.set_page_config(
             page_title="Gramformer Demo",
             initial_sidebar_state="expanded",
@@ -123,7 +125,8 @@ class GramformerDemo:
             st.stop()
 
         with st.spinner('Loading model..'):
-            loadEnModel()
+            if not st.session_state['models_loaded']:
+            	loadEnModel()
             gf = self.load_gf(self.model_map[model_type])
     
         input_text = st.selectbox(