Update app.py

app.py

@@ -11,6 +11,7 @@ def generate_images(prompt: str, num_imgs_to_gen: int):
     Returns:
         all_images (List): List of images that were generated using the model
     """
+    resolution = 512
     dreambooth_model = keras_cv.models.StableDiffusion(
         img_width=resolution, img_height=resolution, jit_compile=True, 
     )