Spaces:

Armandoliv
/

document_parser

Running

Armandoliv commited on Nov 16, 2022

Commit

f9729f3

1 Parent(s): b3565ba

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -136,7 +136,7 @@ def gen_doc(dic_predicted:dict):
   for k,v in dic_predicted.items():
     if "Figure" in k:
-      cv2.imwrite(f'{k}.jpg', dic_predicted['2-Figure'])
       document.add_picture(f'{k}.jpg', width=Inches(3))
     elif "Table" in k:

   for k,v in dic_predicted.items():
     if "Figure" in k:
+      cv2.imwrite(f'{k}.jpg', dic_predicted[k])
       document.add_picture(f'{k}.jpg', width=Inches(3))
     elif "Table" in k: