Only display first page bounding boxes

app.py

@@ -118,7 +118,7 @@ def process_question(question, document, model=list(CHECKPOINTS.keys())[0]):
             # prediction for now
             break
 
-        if "start" in p and "end" in p:
+        if "start" in p and "end" in p and p.get("page") == 0:
             x1, y1, x2, y2 = normalize_bbox(
                 expand_bbox(lift_word_boxes(document)[p["start"] : p["end"] + 1]),
                 image.width,

requirements.txt

@@ -1,4 +1,4 @@
 torch
 git+https://github.com/huggingface/transformers.git@21f6f58721dd9154357576be6de54eefef1f1818
-git+https://github.com/impira/docquery.git@60f743bfa02fb6868da9b6895d5eaf9f8407e776
+git+https://github.com/impira/docquery.git@5de147aa7df285829a8d064f3ca2c7ac74efe0f8
 sentencepiece