torch torchvision transformers gradio Pillow numpy tqdm scikit-learn torchmetrics