Improve example by adding missing import

[avitase]

The example "Creating a Custom Dataset for your files" is missing the import from torch.utils.data import Dataset. Since other imports are shown and the purpose of this example is to show how to create a custom dataset, this import is crucial and should be added.