torch einops basicsr skimage torchvision torchmetrics streamlit