Updated app.py

app.py

@@ -2,7 +2,7 @@ from huggingface_hub import from_pretrained_keras
 from keras_cv import models
 import gradio as gr
 
-dreambooth_model = models.StableDiffusion(img_width=512, img_height=512)
+dreambooth_model = models.StableDiffusion(img_width=256, img_height=256)
 
 diffusion_model = from_pretrained_keras("moizsajid/dreambooth-markhor")
 dreambooth_model._diffusion_model = diffusion_model