Test pre-trained weights

[Indraa145]

Hello! Thank you for the great work, I'm wondering for the pre-trained weight you provide for testing (demo.pt), which dataset did you use to train that model? Thank you!