Update app.py

app.py

@@ -11,7 +11,7 @@ import torch
 from transformers import LayoutLMv3ForTokenClassification
 
 # Load the LayoutLMv3 model
-layout_model = LayoutLMv3ForTokenClassification.from_pretrained("omarelsayeed/trial_simple_random_f")
+layout_model = LayoutLMv3ForTokenClassification.from_pretrained("hantian/layoutreader")
 
 MAX_LEN = 100
 CLS_TOKEN_ID = 0
@@ -55,7 +55,6 @@ def parse_logits(logits: torch.Tensor, length: int) -> List[int]:
 
 def get_orders(image_path, boxes):
     b = scale_and_normalize_boxes(boxes)
-    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