torch scikit-learn transformers Pillow