Output from HF Spaces Demo doesn't match output from code.

[ajkdrag]

Bug description

Output from HF Spaces Demo doesn't match output from code.

Code snippet to reproduce the bug

predictor = ocr_predictor(
        "db_resnet50",
        "crnn_vgg16_bn",
        pretrained=True,
        assume_straight_pages=True,
    export_as_straight_boxes=False
    ).to("cuda:0")

img = cv2.imread(str("myimg.png"), cv2.IMREAD_COLOR)
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
res = predictor([img])

Error traceback

VS

Environment

DocTR version: 0.8.1
TensorFlow version: N/A
PyTorch version: N/A (torchvision N/A)
OpenCV version: N/A
OS: Debian GNU/Linux 11 (bullseye)
Python version: 3.8.18
Is CUDA available (TensorFlow): N/A
Is CUDA available (PyTorch): N/A
CUDA runtime version: 11.8.89
GPU models and configuration: GPU 0: Tesla T4
Nvidia driver version: 525.105.17
cuDNN version: Could not collect

Deep Learning backend

torch = true

[felixdittrich92]

Hi @ajkdrag 👋,
Thanks for the feedback :)
The demo is not up to date with the main branch we have already a ticket to fix this #1090
CC @odulcy-mindee

But the result looks extrem different i will take a look / looks a bit weird 😅

[ajkdrag]

Yes exactly. The demo is giving better results than the main branch. I am not sure if I am doing something wrong . I can share the image privately with you if needed.

[felixdittrich92]

CC @odulcy-mindee
Close because already on track with #1090