Update app.py

app.py

@@ -54,13 +54,11 @@ def parse_logits(logits: torch.Tensor, length: int) -> List[int]:
     return ret
 
 def get_orders(image_path, boxes):
-    print(boxes)
-    boxes = scale_and_normalize_boxes(boxes)
-    print(boxes)
-    inputs = boxes2inputs(boxes)
+    b = scale_and_normalize_boxes(boxes)
+    inputs = boxes2inputs(b)
     inputs = {k: v.to(layout_model.device) for k, v in inputs.items()}  # Move inputs to model device
     logits = layout_model(**inputs).logits.cpu().squeeze(0)  # Perform inference and get logits
-    orders = parse_logits(logits, len(boxes))
+    orders = parse_logits(logits, len(b))
     return orders
 
 
@@ -250,6 +248,7 @@ def remove_overlapping_and_inside_boxes(boxes, classes):
 
 
 def full_predictions(IMAGE_PATH, conf_threshold, iou_threshold):
+    IMAGE_PATH = IMAGE_PATH.resize((1024,1024))
     bboxes, classes = detect_layout(IMAGE_PATH, conf_threshold, iou_threshold)
     bboxes, classes = remove_overlapping_and_inside_boxes(bboxes, classes)
     orders = get_orders(IMAGE_PATH, bboxes)