Update app.py
Browse files
app.py
CHANGED
|
@@ -41,7 +41,7 @@ def detect_layout(img, conf_threshold, iou_threshold):
|
|
| 41 |
return bboxes , classes
|
| 42 |
|
| 43 |
def get_orders(image_path , boxes):
|
| 44 |
-
order_predictions = batch_ordering([image_path], [
|
| 45 |
return [i.position for i in order_predictions[0].bboxes]
|
| 46 |
|
| 47 |
def draw_bboxes_on_image(image_path, bboxes, classes, reading_order):
|
|
|
|
| 41 |
return bboxes , classes
|
| 42 |
|
| 43 |
def get_orders(image_path , boxes):
|
| 44 |
+
order_predictions = batch_ordering([image_path], [boxes], order_model, processor)
|
| 45 |
return [i.position for i in order_predictions[0].bboxes]
|
| 46 |
|
| 47 |
def draw_bboxes_on_image(image_path, bboxes, classes, reading_order):
|