import torchvision

def create_vgg_model():
   
    # Create model
    model_weights = torchvision.models.VGG19_Weights.DEFAULT
    model = torchvision.models.vgg19(weights=model_weights)
    

    # Freeze layers
    for param in model.parameters():
      param.requires_grad = False

    # Kepp only the features of the model
    model = model.features
    
    return model