johko commited on
Commit
91d8eda
1 Parent(s): 48a541f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ resolution = 512
9
  dreambooth_model = keras_cv.models.StableDiffusion(
10
  img_width=resolution, img_height=resolution, jit_compile=True,
11
  )
12
- loaded_diffusion_model = from_pretrained_keras("johko/dreambooth_marvin_paranoid_android")
13
  dreambooth_model._diffusion_model = loaded_diffusion_model
14
 
15
 
 
9
  dreambooth_model = keras_cv.models.StableDiffusion(
10
  img_width=resolution, img_height=resolution, jit_compile=True,
11
  )
12
+ loaded_diffusion_model = from_pretrained_keras("keras-dreambooth/marvin_paranoid_android")
13
  dreambooth_model._diffusion_model = loaded_diffusion_model
14
 
15