Update app.py

app.py

@@ -26,7 +26,7 @@ def predict_reading_order(boxes,image_path):
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     for k,v in encoding.items():
       encoding[k] = v.to(device)
-    outputs = model(**encoding)
+    outputs = reading_order_model(**encoding)
     predictions = outputs.logits.argmax(-1).squeeze().tolist()
     token_boxes = encoding.bbox.squeeze().tolist()
     is_subword = np.array(offset_mapping.squeeze().tolist())[:,0] != 0