Update app.py

app.py

@@ -8,12 +8,11 @@ from paddleocr.ppstructure.recovery.recovery_to_doc import sorted_layout_boxes,
 table_engine = PPStructure(recovery=True)
 # English image
 # table_engine = PPStructure(recovery=True, lang='en')
-def find_layout(name):
+def find_layout(image):
     save_folder = './output'
-    img_path = '0.png'
-    img = cv2.imread(img_path)
+    img = cv2.imread(image)
     result = table_engine(img)
-    save_structure_res(result, save_folder, os.path.basename(img_path).split('.')[0])
+    save_structure_res(result, save_folder, os.path.basename("result").split('.')[0])
 
     final_text = ""
     for line in result:
@@ -22,9 +21,9 @@ def find_layout(name):
         final_text += "\n" + line
     h, w, _ = img.shape
     res = sorted_layout_boxes(result, w)
-    convert_info_docx(img, res, save_folder, os.path.basename(img_path).split('.')[0])
+    convert_info_docx(img, res, save_folder, os.path.basename("result").split('.')[0])
 
     return final_text
 
-iface = gr.Interface(fn=find_layout, inputs="text", outputs="text")
+iface = gr.Interface(fn=find_layout, inputs=[gr.Image()], outputs="text")
 iface.launch()