Spaces:
Runtime error
Runtime error
Johannes
commited on
Commit
Β·
ec1e639
1
Parent(s):
d7f77a0
small update
Browse files
README.md
CHANGED
@@ -2,7 +2,7 @@
|
|
2 |
title: Dreambooth Monkey Island
|
3 |
emoji: π
|
4 |
colorFrom: blue
|
5 |
-
colorTo:
|
6 |
sdk: gradio
|
7 |
sdk_version: 3.23.0
|
8 |
app_file: app.py
|
|
|
2 |
title: Dreambooth Monkey Island
|
3 |
emoji: π
|
4 |
colorFrom: blue
|
5 |
+
colorTo: green
|
6 |
sdk: gradio
|
7 |
sdk_version: 3.23.0
|
8 |
app_file: app.py
|
app.py
CHANGED
@@ -9,7 +9,7 @@ resolution = 512
|
|
9 |
dreambooth_model = keras_cv.models.StableDiffusion(
|
10 |
img_width=resolution, img_height=resolution, jit_compile=True,
|
11 |
)
|
12 |
-
loaded_diffusion_model = from_pretrained_keras("
|
13 |
dreambooth_model._diffusion_model = loaded_diffusion_model
|
14 |
|
15 |
|
@@ -35,11 +35,11 @@ gr.Interface(
|
|
35 |
""",
|
36 |
fn=generate_images,
|
37 |
inputs=[
|
38 |
-
gr.Textbox(label="Positive Prompt"
|
39 |
-
gr.Textbox(label="Negative Prompt", value="
|
40 |
gr.Slider(label='Number of gen image', minimum=1, maximum=4, value=2, step=1),
|
41 |
-
gr.Slider(label="Inference Steps", value=
|
42 |
-
gr.Slider(label='Guidance scale', value=7
|
43 |
],
|
44 |
outputs=[
|
45 |
gr.Gallery(show_label=False).style(grid=(1,2)),
|
|
|
9 |
dreambooth_model = keras_cv.models.StableDiffusion(
|
10 |
img_width=resolution, img_height=resolution, jit_compile=True,
|
11 |
)
|
12 |
+
loaded_diffusion_model = from_pretrained_keras("keras-dreambooth/monkey_island_style")
|
13 |
dreambooth_model._diffusion_model = loaded_diffusion_model
|
14 |
|
15 |
|
|
|
35 |
""",
|
36 |
fn=generate_images,
|
37 |
inputs=[
|
38 |
+
gr.Textbox(label="Positive Prompt"),
|
39 |
+
gr.Textbox(label="Negative Prompt", value="bad, ugly"),
|
40 |
gr.Slider(label='Number of gen image', minimum=1, maximum=4, value=2, step=1),
|
41 |
+
gr.Slider(label="Inference Steps", value=70),
|
42 |
+
gr.Slider(label='Guidance scale', value=7, maximum=15, minimum=0, step=0.5),
|
43 |
],
|
44 |
outputs=[
|
45 |
gr.Gallery(show_label=False).style(grid=(1,2)),
|