merve HF Staff commited on
Commit
cb492a7
·
1 Parent(s): cf5cbd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("merve/dreambooth_bioshock")
9
  sd_dreambooth_model._diffusion_model = db_diffusion_model
10
 
11
  # generate images
@@ -35,5 +35,5 @@ gr.Interface(
35
  ],
36
  title="Dreambooth Bioshock",
37
  description = "This is a dreambooth model fine-tuned on images of Bioshock sceneries. To play with the demo, input the concept with {sks bskscnry}.",
38
- examples = [["a city in sks bskscnry style, rendered in unreal engine, trending on art station", "", 2, 100, 7.5]],
39
  cache_examples=True).launch()
 
5
  sd_dreambooth_model = models.StableDiffusion(
6
  img_width=512, img_height=512
7
  )
8
+ db_diffusion_model = from_pretrained_keras("merve/dreambooth_bioshock_v2")
9
  sd_dreambooth_model._diffusion_model = db_diffusion_model
10
 
11
  # generate images
 
35
  ],
36
  title="Dreambooth Bioshock",
37
  description = "This is a dreambooth model fine-tuned on images of Bioshock sceneries. To play with the demo, input the concept with {sks bskscnry}.",
38
+ examples = [["a cafe in sks bskscnry style, rendered in unreal engine, trending on art station", "", 2, 100, 7.5]],
39
  cache_examples=True).launch()