Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -385,7 +385,7 @@ def process_image(image):
|
|
| 385 |
expand_rowcol_bbox_bottom = 0
|
| 386 |
|
| 387 |
image = image.convert("RGB")
|
| 388 |
-
model, probas, bboxes_scaled = table_detector(image, THRESHOLD_PROBA=
|
| 389 |
# plot_results_detection(model, image, probas, bboxes_scaled, delta_xmin, delta_ymin, delta_xmax, delta_ymax)
|
| 390 |
cropped_img_list = crop_tables(
|
| 391 |
image, probas, bboxes_scaled, delta_xmin, delta_ymin, delta_xmax, delta_ymax
|
|
@@ -397,7 +397,7 @@ def process_image(image):
|
|
| 397 |
unpadded_table, padd_top, padd_right, padd_bottom, padd_left
|
| 398 |
)
|
| 399 |
model, probas, bboxes_scaled = table_struct_recog(
|
| 400 |
-
table, THRESHOLD_PROBA=
|
| 401 |
)
|
| 402 |
rows, cols = generate_structure(
|
| 403 |
model,
|
|
@@ -444,3 +444,4 @@ iface = gr.Interface(
|
|
| 444 |
)
|
| 445 |
iface.launch(debug=True)
|
| 446 |
|
|
|
|
|
|
| 385 |
expand_rowcol_bbox_bottom = 0
|
| 386 |
|
| 387 |
image = image.convert("RGB")
|
| 388 |
+
model, probas, bboxes_scaled = table_detector(image, THRESHOLD_PROBA=TD_THRESHOLD)
|
| 389 |
# plot_results_detection(model, image, probas, bboxes_scaled, delta_xmin, delta_ymin, delta_xmax, delta_ymax)
|
| 390 |
cropped_img_list = crop_tables(
|
| 391 |
image, probas, bboxes_scaled, delta_xmin, delta_ymin, delta_xmax, delta_ymax
|
|
|
|
| 397 |
unpadded_table, padd_top, padd_right, padd_bottom, padd_left
|
| 398 |
)
|
| 399 |
model, probas, bboxes_scaled = table_struct_recog(
|
| 400 |
+
table, THRESHOLD_PROBA=TSR_THRESHOLD
|
| 401 |
)
|
| 402 |
rows, cols = generate_structure(
|
| 403 |
model,
|
|
|
|
| 444 |
)
|
| 445 |
iface.launch(debug=True)
|
| 446 |
|
| 447 |
+
|