Onur Savas commited on
Commit
6afb5fb
·
1 Parent(s): abadd53

updated app

Browse files
Files changed (1) hide show
  1. app.py +13 -14
app.py CHANGED
@@ -8,23 +8,22 @@ from paddleocr.ppstructure.recovery.recovery_to_doc import sorted_layout_boxes,
8
  table_engine = PPStructure(recovery=True)
9
  # English image
10
  # table_engine = PPStructure(recovery=True, lang='en')
 
 
 
 
 
 
11
 
12
- save_folder = './output'
13
- img_path = '0.png'
14
- img = cv2.imread(img_path)
15
- result = table_engine(img)
16
- save_structure_res(result, save_folder, os.path.basename(img_path).split('.')[0])
17
 
18
- for line in result:
19
- line.pop('img')
20
- print(line)
21
 
22
- h, w, _ = img.shape
23
- res = sorted_layout_boxes(result, w)
24
- convert_info_docx(img, res, save_folder, os.path.basename(img_path).split('.')[0])
25
-
26
- def greet(name):
27
- return "Hello " + name + "!!"
28
 
29
  iface = gr.Interface(fn=find_layout, inputs="text", outputs="text")
30
  iface.launch()
 
8
  table_engine = PPStructure(recovery=True)
9
  # English image
10
  # table_engine = PPStructure(recovery=True, lang='en')
11
+ def find_layout(name):
12
+ save_folder = './output'
13
+ img_path = '0.png'
14
+ img = cv2.imread(img_path)
15
+ result = table_engine(img)
16
+ save_structure_res(result, save_folder, os.path.basename(img_path).split('.')[0])
17
 
18
+ for line in result:
19
+ line.pop('img')
20
+ print(line)
 
 
21
 
22
+ h, w, _ = img.shape
23
+ res = sorted_layout_boxes(result, w)
24
+ convert_info_docx(img, res, save_folder, os.path.basename(img_path).split('.')[0])
25
 
26
+ return "Done processing 0.png"
 
 
 
 
 
27
 
28
  iface = gr.Interface(fn=find_layout, inputs="text", outputs="text")
29
  iface.launch()