Update app.py

app.py

@@ -4,7 +4,7 @@ from huggingface_hub import snapshot_download
 from PIL import Image
 from PIL import Image, ImageDraw, ImageFont
 
-
+import random
 from collections import defaultdict
 from typing import List, Dict
 import torch
@@ -55,6 +55,7 @@ def parse_logits(logits: torch.Tensor, length: int) -> List[int]:
 
 def get_orders(image_path, boxes):
     b = scale_and_normalize_boxes(boxes)
+    b = random.shuffle(b)
     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