torch torchvision transformers gradio diffusers datasets scikit-learn Pillow