Update pages/type_text.py

pages/type_text.py

@@ -7,7 +7,10 @@ from transformers import pipeline # AutoTokenizer, AutoModelForCausalLM, AutoMod
 from sentence_transformers import SentenceTransformer, util 
 
 import os
-os.getenv("HF_TOKEN")
+os.getenv("HF_TOKEN") 
+
+for k, v in st.session_state.items():
+    st.session_state[k] = v
 
 def get_device_map() -> str:
     return 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -38,7 +41,7 @@ numMAPPINGS_input = 5
 #numMAPPINGS_input = st.text_input("Type number of mappings and hit Enter", key="user_input_numMAPPINGS")
 #st.button("Clear text", on_click=on_click)
 
-st.write("ZZZZZZZZZZZZ: ", chapter_names)
+st.write("ZZZZZZZZZZZZ: ", selected_chapter_names)
 
 @st.cache_resource
 def load_model():