Update app.py

app.py

@@ -1,11 +1,11 @@
 from huggingface_hub import from_pretrained_keras
-import keras_cv
+from keras_cv import models
 import gradio as gr
 
 
 # prepare model
 resolution = 512
-sd_dreambooth_model = keras_cv.models.StableDiffusion(
+sd_dreambooth_model = models.StableDiffusion(
     img_width=resolution, img_height=resolution, jit_compile=True, 
 )
 db_diffusion_model = from_pretrained_keras("merve/dreambooth_diffusion_model")