import tensorflow as tf

# Load the model
model = tf.keras.models.load_model("model.h5")

# Generate the image
image = model.predict(text_prompt)

# Save the image
tf.keras.utils.save_img(image, "image.png")