Hugging Face's logo

Spaces:
keras-dreambooth
/
dreambooth_teddy
Runtime error

App Files Community
danwaldie commited on
Commit
eafc480
•
1 Parent(s): f90fd66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import gradio as gr
5
  sd_dreambooth_model = models.StableDiffusion(
6
  img_width=512, img_height=512
7
  )
8
- db_diffusion_model = from_pretrained_keras("dan/dreambooth_teddy")
9
  sd_dreambooth_model._diffusion_model = db_diffusion_model
10
 
11
  # generate images
 
5
  sd_dreambooth_model = models.StableDiffusion(
6
  img_width=512, img_height=512
7
  )
8
+ db_diffusion_model = from_pretrained_keras("danwaldie/dreambooth_teddy")
9
  sd_dreambooth_model._diffusion_model = db_diffusion_model
10
 
11
  # generate images