# -*- coding: utf-8 -*-
"""generationImageLearning.ipynb

Automatically generated by Colab.

Original file is located at
    https://colab.research.google.com/drive/1wL1Te0FkPTDa9t8YTa1jeeBVIFyf4tXV
"""

!pip install tensorflow keras_cv --upgrade --quiet

!pip install pycocotools

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 in a dark alley with purple lights", batch_size=4)


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)