torch torchvision numpy scikit-learn pillow matplotlib gradio joblib