Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -370,21 +370,20 @@ def create_dataframe(cells_pytess_result: list, max_cols: int, max_rows: int, cs
|
|
370 |
return df
|
371 |
|
372 |
|
373 |
-
def process_image(
|
374 |
-
|
375 |
-
|
376 |
-
|
377 |
-
|
378 |
-
|
379 |
-
|
380 |
-
|
381 |
-
|
382 |
-
|
383 |
-
|
384 |
-
|
385 |
-
|
386 |
-
|
387 |
-
):
|
388 |
image = image.convert("RGB")
|
389 |
model, probas, bboxes_scaled = table_detector(image, THRESHOLD_PROBA=td_threshold)
|
390 |
# plot_results_detection(model, image, probas, bboxes_scaled, delta_xmin, delta_ymin, delta_xmax, delta_ymax)
|
@@ -436,21 +435,7 @@ examples = [["image_0.png"]]
|
|
436 |
|
437 |
iface = gr.Interface(
|
438 |
fn=process_image,
|
439 |
-
inputs=
|
440 |
-
gr.Image(type="pil"),
|
441 |
-
gr.Slider(0, 1, 0.9),
|
442 |
-
gr.Slider(0, 1, 0.8),
|
443 |
-
gr.Slider(0, 200, 100),
|
444 |
-
gr.Slider(0, 200, 100),
|
445 |
-
gr.Slider(0, 200, 100),
|
446 |
-
gr.Slider(0, 200, 100),
|
447 |
-
gr.Number(0),
|
448 |
-
gr.Number(0),
|
449 |
-
gr.Number(0),
|
450 |
-
gr.Number(0),
|
451 |
-
gr.Number(0),
|
452 |
-
gr.Number(0),
|
453 |
-
],
|
454 |
outputs="text",
|
455 |
title=title,
|
456 |
description=description,
|
|
|
370 |
return df
|
371 |
|
372 |
|
373 |
+
def process_image(image):
|
374 |
+
TD_THRESHOLD = 0.9
|
375 |
+
TSR_THRESHOLD = 0.8
|
376 |
+
padd_top = 100
|
377 |
+
padd_left = 100
|
378 |
+
padd_bottom = 100
|
379 |
+
padd_right = 20
|
380 |
+
delta_xmin = 0
|
381 |
+
delta_ymin = 0
|
382 |
+
delta_xmax = 0
|
383 |
+
delta_ymax = 0
|
384 |
+
expand_rowcol_bbox_top = 0
|
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)
|
|
|
435 |
|
436 |
iface = gr.Interface(
|
437 |
fn=process_image,
|
438 |
+
inputs=gr.Image(type="pil"),
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
439 |
outputs="text",
|
440 |
title=title,
|
441 |
description=description,
|