model.py

from tensorflow.keras.models import load_model

# Load your model
model = load_model("my_image_model.h5")

# Print model summary
model.summary()