import torch import torchvision.models as models from torchvision import transforms from PIL import Image
