Johannes commited on
Commit
ec1e639
Β·
1 Parent(s): d7f77a0

small update

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +5 -5
README.md CHANGED
@@ -2,7 +2,7 @@
2
  title: Dreambooth Monkey Island
3
  emoji: πŸ’
4
  colorFrom: blue
5
- colorTo: indigo
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("johko/monkey_island_style")
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", value="a photo of paranoid marvin a robot"),
39
- gr.Textbox(label="Negative Prompt", value="low quality, deformed"),
40
  gr.Slider(label='Number of gen image', minimum=1, maximum=4, value=2, step=1),
41
- gr.Slider(label="Inference Steps", value=50),
42
- gr.Slider(label='Guidance scale', value=7.5, maximum=15, minimum=0, step=0.5),
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)),