import torch
from torchvision.models import resnet50, ResNet50_Weights