add custom dataset for training.

[Captain272]

This piece of code facilitates training the current model with the custom dataset.