Update pages/type_text.py

pages/type_text.py

@@ -16,7 +16,7 @@ for k, v in st.session_state.items():
 #selected_chapters_intlist = [int(i) for i in selected_chapters_floatlist]
 #st.write("SELECTED CHAPTERS: ", selected_chapters_intlist)
 selected_rows_list = list(st.session_state.items())[1][1]
-st.write("SELECTED ROWS: ", selected_rows_list)
+#st.write("SELECTED ROWS: ", selected_rows_list)
 
 def get_device_map() -> str:
     return 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -73,7 +73,8 @@ INTdesc_embedding = model.encode(INTdesc_input)
 if len(selected_rows_list) == 0: 
     st.warning("Please select at least one chapter") 
     selected_rows_list = [0, 10080]
-#df_SBS = pd.read_csv("SBS_V2_0/Code_Table.csv", header=0, skip_blank_lines=False, skiprows = lambda x: x not in selected_rows_list)
+    st.write("SELECTED ROWS: ", selected_rows_list)
+df_SBS = pd.read_csv("SBS_V2_0/Code_Table.csv", header=0, skip_blank_lines=False, skiprows = lambda x: x not in selected_rows_list)
 
 #df_SBS = pd.read_csv("SBS_V2_0/Code_Table.csv", index_col="SBS_Code", usecols=["Long_Description"]) # na_values=['NA']
 #df_SBS = pd.read_csv("SBS_V2_0/Code_Table.csv", usecols=["SBS_Code_Hyphenated","Long_Description"])