Update app.py

app.py

@@ -163,22 +163,22 @@ def gen_doc(dic_predicted:dict):
 def main_convert(filename):
   print(filename.name)
   doc = to_image(filename.name)
-  print("doc",type(doc))
+
   img, detected = detect(doc)
-  print("img ", type(img))
+
   n_detected = get_detected(img, detected)
-  print("n_detected ",type(n_detected ))
+
   dic_predicted = predict_elements(img, n_detected)
-  print("dic_predicted ",type(dic_predicted ), dic_predicted.keys() )
+
   gen_doc(dic_predicted)
-  print("File Generated")
 
   im_out = lp.draw_box(img, detected, box_width=5, box_alpha=0.2, show_element_type=True)
-  print("Keys final")
-  print(dic_predicted.keys())
-  print(f"img_out {im_out}")
-  print(f"{im_out.shape}")
-  return  'demo.docx', im_out, dic_predicted
+  dict_out = {}
+  for k,v in dic_predicted.items():
+    if "figure" not in k.lower():
+      dict_out[k] = dic_predicted[k]
+
+  return  'demo.docx', im_out, dict_out
   
   
 inputs = [gr.File(type='file', label="Original PDF File")]