Minor changes

app.py

@@ -83,7 +83,7 @@ def table_detection(pil_img, imgsz=640):
     image = PIL_to_cv(pil_img)
     pred = detection_model(image, size=imgsz)
     pred = pred.xywhn[0]
-    result = pred.cpu().numpy()
+    result = pred.detach().cpu().numpy()
     return result
 
 
@@ -91,7 +91,7 @@ def table_structure(pil_img, imgsz=640):
     image = PIL_to_cv(pil_img)
     pred = structure_model(image, size=imgsz)
     pred = pred.xywhn[0]
-    result = pred.cpu().numpy()
+    result = pred.detach().cpu().numpy()
     return result
 
 
@@ -211,7 +211,7 @@ def convert_stucture(page_tokens, pil_img, structure_result):
     # print('table_class_objects:', table_class_objects)
     # print('table_bbox:', table_bbox)
 
-    tokens_in_table = [token for token in page_tokens if postprocess.iob(token['bbox'], table_bbox) >= 0.01]
+    tokens_in_table = [token for token in page_tokens if postprocess.iob(token['bbox'], table_bbox) >= 0.001]
     # print('tokens_in_table:', tokens_in_table)
 
     table_structures, cells, confidence_score = postprocess.objects_to_cells(table, table_objects, tokens_in_table, structure_class_names, structure_class_thresholds)