SWHL commited on
Commit
548d9af
·
verified ·
1 Parent(s): eec57a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -76,7 +76,8 @@ def get_ocr_result(
76
  )
77
  vis_img = ocr_result.vis()
78
  if return_word_box:
79
- txts, scores, _ = list(zip(*ocr_result.word_results))
 
80
  ocr_txts = [[i, txt, score] for i, (txt, score) in enumerate(zip(txts, scores))]
81
  return vis_img, ocr_txts, ocr_result.elapse
82
 
 
76
  )
77
  vis_img = ocr_result.vis()
78
  if return_word_box:
79
+ full_word_results = sum(ocr_result.word_results, ())
80
+ txts, scores, _ = list(zip(*full_word_results))
81
  ocr_txts = [[i, txt, score] for i, (txt, score) in enumerate(zip(txts, scores))]
82
  return vis_img, ocr_txts, ocr_result.elapse
83