Hugging Face's logo

Spaces:
keras-dreambooth
/
dreambooth-markhor
Runtime error

App Files Community
moizsajid commited on
Commit
e01698d
·
1 Parent(s): 4fc79e2

Updated app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ from huggingface_hub import from_pretrained_keras
2
  from keras_cv import models
3
  import gradio as gr
4
 
5
- dreambooth_model = models.StableDiffusion(img_width=512, img_height=512)
6
 
7
  diffusion_model = from_pretrained_keras("moizsajid/dreambooth-markhor")
8
  dreambooth_model._diffusion_model = diffusion_model
 
2
  from keras_cv import models
3
  import gradio as gr
4
 
5
+ dreambooth_model = models.StableDiffusion(img_width=256, img_height=256)
6
 
7
  diffusion_model = from_pretrained_keras("moizsajid/dreambooth-markhor")
8
  dreambooth_model._diffusion_model = diffusion_model