Spaces:

keras-dreambooth
/

example-submission

Running

osanseviero commited on Mar 2, 2023

Commit

7bbe940

1 Parent(s): 693bcea

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ resolution = 512
 sd_dreambooth_model = models.StableDiffusion(
     img_width=resolution, img_height=resolution
 )
-db_diffusion_model = from_pretrained_keras("merve/dreambooth_diffusion_model")
 sd_dreambooth_model._diffusion_model = db_diffusion_model
 # generate images

 sd_dreambooth_model = models.StableDiffusion(
     img_width=resolution, img_height=resolution
 )
+db_diffusion_model = from_pretrained_keras("keras-dreambooth/dreambooth_diffusion_model")
 sd_dreambooth_model._diffusion_model = db_diffusion_model
 # generate images