shivi commited on
Commit
7d610b2
1 Parent(s): dbb8989

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ resolution = 512
8
  dreambooth_model = keras_cv.models.StableDiffusion(
9
  img_width=resolution, img_height=resolution, jit_compile=True,
10
  )
11
- loaded_diffusion_model = from_pretrained_keras("shivi/dreambooth_diffusion_model")
12
  dreambooth_model._diffusion_model = loaded_diffusion_model
13
 
14
 
 
8
  dreambooth_model = keras_cv.models.StableDiffusion(
9
  img_width=resolution, img_height=resolution, jit_compile=True,
10
  )
11
+ loaded_diffusion_model = from_pretrained_keras("keras-dreambooth/pink-floyd-division-bell")
12
  dreambooth_model._diffusion_model = loaded_diffusion_model
13
 
14