The demonstration dataset used was dataset A of https://doi.org/10.5281/zenodo.11653626
Only a small subset of the test dataset and a pre-trained model is used in `Demo only inference.ipynb`, so that all the data can be shipped in the repository for a quick example.
The `Demo.ipynb` notebook does a full training with the training dataset A in https://doi.org/10.5281/zenodo.11653626