onursavas's picture
Update app.py
a660df8
raw
history blame contribute delete
629 Bytes
import gradio as gr
import os
import numpy as np
from paddleocr import PPStructure,draw_structure_result,save_structure_res
# Chinese image
table_engine = PPStructure(show_log=True)
examples = ['0.png']
def find_layout(image):
img = np.array(image)
result = table_engine(img)
# save_structure_res(result, save_folder,os.path.basename(img_path).split('.')[0])
output = []
for line in result:
line.pop('img')
print(line)
output.append(line)
return output
iface = gr.Interface(fn=find_layout, inputs=[gr.Image(type="pil")], outputs="text", examples=examples)
iface.launch()