Update model_t.py

model_t.py

@@ -0,0 +1,19 @@
+import time
+import keras_cv
+from tensorflow import keras
+import matplotlib.pyplot as plt
+
+model = keras_cv.models.StableDiffusion(img_width=512, img_height=512)
+
+images = model.text_to_image("photograph of an astronaut riding a horse", batch_size=3)
+
+
+def plot_images(images):
+    plt.figure(figsize=(20, 20))
+    for i in range(len(images)):
+        ax = plt.subplot(1, len(images), i + 1)
+        plt.imshow(images[i])
+        plt.axis("off")
+
+
+plot_images(images)