georad commited on
Commit
cbb5134
·
verified ·
1 Parent(s): b32390e

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +2 -2
pages/type_text.py CHANGED
@@ -12,8 +12,8 @@ os.getenv("HF_TOKEN")
12
 
13
  for k, v in st.session_state.items():
14
  st.session_state[k] = v
15
- selected_chapters_listfloat = list(st.session_state.items())[0][1]
16
- selected_chapters_list = int(selected_chapters_listfloat)
17
 
18
  def get_device_map() -> str:
19
  return 'cuda' if torch.cuda.is_available() else 'cpu'
 
12
 
13
  for k, v in st.session_state.items():
14
  st.session_state[k] = v
15
+ selected_chapters_floatlist = list(st.session_state.items())[0][1]
16
+ selected_chapters_list = [int(i) for i in selected_chapters_floatlist]
17
 
18
  def get_device_map() -> str:
19
  return 'cuda' if torch.cuda.is_available() else 'cpu'