Spaces:

georad
/

sbsmapper

Running

georad commited on May 1

Commit

d54e4b9

verified ·

1 Parent(s): 35869d3

Update pages/type_text.py

Files changed (1) hide show

pages/type_text.py CHANGED Viewed

@@ -13,7 +13,7 @@ os.getenv("HF_TOKEN")
 for k, v in st.session_state.items():
     st.session_state[k] = v
-st.write("SSSSSSSSSSSSSSSSSS: ", st.session_state.items()["selected_chapters"])
 def get_device_map() -> str:
     return 'cuda' if torch.cuda.is_available() else 'cpu'

 for k, v in st.session_state.items():
     st.session_state[k] = v
+st.write("SSSSSSSSSSSSSSSSSS: ", list(st.session_state.items())[0])
 def get_device_map() -> str:
     return 'cuda' if torch.cuda.is_available() else 'cpu'