from tensorflow.keras.applications import VGG16 def create_model(input_shape=(224, 224, 3), num_classes=10): return VGG16(weights=None, include_top=True, input_shape=input_shape, classes=num_classes)