SengTak commited on
Commit
a0afa64
·
1 Parent(s): d09b84c
Files changed (1) hide show
  1. app_pages/ocr_comparator.py +5 -4
app_pages/ocr_comparator.py CHANGED
@@ -1375,9 +1375,10 @@ def app():
1375
  if not st.session_state.df_results.empty:
1376
  ##----------- Show recognition results ------------------------------------------------------------
1377
  results_cols = st.session_state.df_results.columns
1378
- list_col_text = np.arange(1, len(cols_size)-1, 2)
1379
- list_col_confid = np.arange(2, len(cols_size)-1, 2)
1380
-
 
1381
  dict_draw_reco = {'in_image': st.session_state.list_images[1], \
1382
  'in_boxes_coordinates': st.session_state.list_boxes, \
1383
  'in_list_texts': [st.session_state.df_results[x].to_list() \
@@ -1386,7 +1387,7 @@ def app():
1386
  for x in results_cols[list_col_confid]], \
1387
  'in_dict_back_colors': dict_back_colors, \
1388
  'in_df_results_tesseract' : st.session_state.df_results_tesseract, \
1389
- 'in_reader_type_list': reader_type_list
1390
  }
1391
  show_reco = st.empty()
1392
 
 
1375
  if not st.session_state.df_results.empty:
1376
  ##----------- Show recognition results ------------------------------------------------------------
1377
  results_cols = st.session_state.df_results.columns
1378
+ list_col_text = np.arange(1, len(cols_size)-2, 2)
1379
+ list_col_confid = np.arange(2, len(cols_size)-2, 2)
1380
+ reader_type_list_cp = reader_type_list.copy()
1381
+ reader_type_list_cp.remove("PPOCR")
1382
  dict_draw_reco = {'in_image': st.session_state.list_images[1], \
1383
  'in_boxes_coordinates': st.session_state.list_boxes, \
1384
  'in_list_texts': [st.session_state.df_results[x].to_list() \
 
1387
  for x in results_cols[list_col_confid]], \
1388
  'in_dict_back_colors': dict_back_colors, \
1389
  'in_df_results_tesseract' : st.session_state.df_results_tesseract, \
1390
+ 'in_reader_type_list': reader_type_list_cp
1391
  }
1392
  show_reco = st.empty()
1393