omarelsayeed commited on
Commit
555ccca
·
verified ·
1 Parent(s): 69dd8ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def parse_logits(logits: torch.Tensor, length: int) -> List[int]:
55
 
56
  def get_orders(image_path, boxes):
57
  b = scale_and_normalize_boxes(boxes)
58
- b = random.shuffle(b)
59
  inputs = boxes2inputs(b)
60
  inputs = {k: v.to(layout_model.device) for k, v in inputs.items()} # Move inputs to model device
61
  logits = layout_model(**inputs).logits.cpu().squeeze(0) # Perform inference and get logits
 
55
 
56
  def get_orders(image_path, boxes):
57
  b = scale_and_normalize_boxes(boxes)
58
+ random.shuffle(b)
59
  inputs = boxes2inputs(b)
60
  inputs = {k: v.to(layout_model.device) for k, v in inputs.items()} # Move inputs to model device
61
  logits = layout_model(**inputs).logits.cpu().squeeze(0) # Perform inference and get logits