added hf compatibility

[snova-guhae]

[czczup]

Hello, thank you for contributing to our repository. If you wish to have your code merged, I suggest keeping the original data loading method as the default option, with the hf load_dataset as an alternative.