georad commited on
Commit
b2b05d5
·
verified ·
1 Parent(s): e74e4bd

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +4 -1
pages/type_text.py CHANGED
@@ -10,11 +10,14 @@ import os
10
  os.getenv("HF_TOKEN")
11
 
12
  for k, v in st.session_state.items():
13
- st.session_state[k] = v
 
14
  selected_chapters_floatlist = list(st.session_state.items())[0][1]
15
  selected_chapters_intlist = [int(i) for i in selected_chapters_floatlist]
16
  st.write("SELECTED CHAPTERS: ", selected_chapters_intlist)
17
 
 
 
18
  def get_device_map() -> str:
19
  return 'cuda' if torch.cuda.is_available() else 'cpu'
20
  device = get_device_map() # 'cpu'
 
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
 
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'
23
  device = get_device_map() # 'cpu'