Update dataset.py

Author: ExponentialML

utils/dataset.py

@@ -7,6 +7,7 @@
 import torchvision.transforms as T
 import torch
 
+from glob import glob
 from PIL import Image
 from itertools import islice
 from pathlib import Path