Update app.py
Browse files
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/
|
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
|
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()
|