georad commited on
Commit
33e88f5
·
verified ·
1 Parent(s): 7e22405

Update pages/type_text.py

Browse files
Files changed (1) hide show
  1. pages/type_text.py +4 -5
pages/type_text.py CHANGED
@@ -14,7 +14,6 @@ for k, v in st.session_state.items():
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
- st.write("SELECTED CHAPTERS FROM PAGE 1: ", selected_chapters_list)
18
 
19
  def get_device_map() -> str:
20
  return 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -63,10 +62,10 @@ INTdesc_embedding = model.encode(INTdesc_input)
63
 
64
  #df_SBS = pd.read_csv("SBS_V2_Table.csv", index_col="SBS_Code", usecols=["Long_Description"]) # na_values=['NA']
65
  #df_SBS = pd.read_csv("SBS_V2_Table.csv", usecols=["SBS_Code_Hyphenated","Long_Description"])
66
- from_line = 7725 # Imaging services chapter start, adjust as needed
67
- to_line = 8239 # Imaging services chapter end, adjust as needed
68
- nrows = to_line - from_line + 1
69
- skiprows = list(range(1,from_line - 1))
70
  df_SBS = pd.read_csv("SBS_V2_Table.csv", header=0, skip_blank_lines=False, skiprows=skiprows, nrows=nrows)
71
  #st.write(df_SBS.head(5))
72
 
 
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'
 
62
 
63
  #df_SBS = pd.read_csv("SBS_V2_Table.csv", index_col="SBS_Code", usecols=["Long_Description"]) # na_values=['NA']
64
  #df_SBS = pd.read_csv("SBS_V2_Table.csv", usecols=["SBS_Code_Hyphenated","Long_Description"])
65
+ from_row_index = 7725 # Imaging services chapter start, adjust as needed
66
+ to_row_index = 8239 # Imaging services chapter end, adjust as needed
67
+ nrows = to_row_index - from_row_index + 1
68
+ skiprows = list(range(1,from_row_index - 1))
69
  df_SBS = pd.read_csv("SBS_V2_Table.csv", header=0, skip_blank_lines=False, skiprows=skiprows, nrows=nrows)
70
  #st.write(df_SBS.head(5))
71