onursavas commited on
Commit
c7f2632
·
1 Parent(s): 60e5f27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -15,11 +15,11 @@ table_engine = PPStructure(recovery=True)
15
  examples = ['0.png']
16
 
17
  def find_layout(image):
18
- save_folder = './output'
19
  # img = cv2.imread(image)
20
  img = np.array(image)
21
  result = table_engine(img)
22
- save_structure_res(result, save_folder, os.path.basename("result").split('.')[0])
23
 
24
  final_text = []
25
  for line in result:
@@ -27,9 +27,9 @@ def find_layout(image):
27
  print(line)
28
  final_text.append(line)
29
 
30
- h, w, _ = img.shape
31
- res = sorted_layout_boxes(result, w)
32
- convert_info_docx(img, res, save_folder, os.path.basename("result").split('.')[0])
33
 
34
  return final_text
35
 
 
15
  examples = ['0.png']
16
 
17
  def find_layout(image):
18
+ # save_folder = './output'
19
  # img = cv2.imread(image)
20
  img = np.array(image)
21
  result = table_engine(img)
22
+ # save_structure_res(result, save_folder, os.path.basename("result").split('.')[0])
23
 
24
  final_text = []
25
  for line in result:
 
27
  print(line)
28
  final_text.append(line)
29
 
30
+ # h, w, _ = img.shape
31
+ # res = sorted_layout_boxes(result, w)
32
+ # convert_info_docx(img, res, save_folder, os.path.basename("result").split('.')[0])
33
 
34
  return final_text
35