danwaldie
/

dreambooth_teddy

keras-dreambooth

Model card Files Files and versions

danwaldie commited on Mar 12, 2023

Commit

067a549

·

1 Parent(s): 8bc1769

Update app.py

Files changed (1) hide show

app.py +1 -1

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import gradio as gr
 sd_dreambooth_model = models.StableDiffusion(
     img_width=512, img_height=512
 )
-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=512, img_height=512
 )
+db_diffusion_model = from_pretrained_keras("danwaldie/dreambooth_teddy")
 sd_dreambooth_model._diffusion_model = db_diffusion_model
 # generate images