georad commited on
Commit
2b6ebbd
·
verified ·
1 Parent(s): 202bd1c

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +3 -3
pages/type_text.py CHANGED
@@ -9,14 +9,14 @@ from sentence_transformers import SentenceTransformer, util
9
  import os
10
  os.getenv("HF_TOKEN")
11
 
12
- for k, v in st.session_state.items():
13
- st.session_state[k] = v
14
 
15
  #selected_chapters_floatlist = list(st.session_state.items())[0][1]
16
  #selected_chapters_intlist = [int(i) for i in selected_chapters_floatlist]
17
  #st.write("SELECTED CHAPTERS: ", selected_chapters_intlist)
18
  selected_rows_list = list(st.session_state.items())[1][1]
19
- #st.write("SELECTED ROWS: ", selected_rows_list)
20
 
21
  def get_device_map() -> str:
22
  return 'cuda' if torch.cuda.is_available() else 'cpu'
 
9
  import os
10
  os.getenv("HF_TOKEN")
11
 
12
+ #for k, v in st.session_state.items():
13
+ # st.session_state[k] = v
14
 
15
  #selected_chapters_floatlist = list(st.session_state.items())[0][1]
16
  #selected_chapters_intlist = [int(i) for i in selected_chapters_floatlist]
17
  #st.write("SELECTED CHAPTERS: ", selected_chapters_intlist)
18
  selected_rows_list = list(st.session_state.items())[1][1]
19
+ st.write("SELECTED ROWS: ", selected_rows_list)
20
 
21
  def get_device_map() -> str:
22
  return 'cuda' if torch.cuda.is_available() else 'cpu'