utils/preprocess.py

from torchvision import transforms
from PIL import Image

transform = transforms.Compose([
    transforms.Resize((528,528)),
    transforms.ToTensor(),
    transforms.Normalize(
        mean=[0.485,0.456,0.406],
        std=[0.229,0.224,0.225]
    )
])

def preprocess_image(uploaded_file):

    image = Image.open(uploaded_file).convert("RGB")

    tensor = transform(image).unsqueeze(0)

    return image, tensor