torchvision pytorch_lightning==1.7 pandas matplotlib torchmetrics==0.11.4 gradio