Update app.py

app.py

@@ -6,7 +6,7 @@ import matplotlib.pyplot as plt
 from translate import Translator
 
 keras.mixed_precision.set_global_policy("mixed_float16")
-model = keras_cv.models.StableDiffusion(img_width=512, img_height=512, jit_compile=True)
+model = keras_cv.models.StableDiffusion(img_width=256, img_height=256, jit_compile=True)
 
 def plot_images(images):
     plt.figure(figsize=(10, 10))
@@ -16,7 +16,6 @@ def plot_images(images):
         plt.axis("off")
         plt.tight_layout()
 
-images = model.text_to_image("photograph of an astronaut riding a horse", batch_size=3)
 translator = Translator(from_lang="ko", to_lang="en")
 
 def generate_images(text, n=3):