Tensor size mismatch for table detection in PDF

[sanakashgouli]

Hi @NielsRogge,

Thanks for the great work on the table transformer notebook.

I've been working on modifying the code in the notebook to detect tables in a PDF (treating every page as an image), however, for the structure recognition part I encounter the following error:

RuntimeError: The size of tensor a (125) must match the size of tensor b (4) at non-singleton dimension 2.

Could you kindly guide me on what the problem might be?

Thanks!