dgoot commited on
Commit
1789e44
·
1 Parent(s): f5f8b8f

Add stable-diffusion-3 model

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -5,10 +5,12 @@ from loguru import logger
5
  from PIL import Image
6
 
7
  models = [
8
- "stabilityai/stable-diffusion-xl-refiner-1.0",
9
  "stabilityai/sdxl-turbo",
 
 
10
  "timbrooks/instruct-pix2pix",
11
  ]
 
12
 
13
 
14
  @logger.catch(reraise=True)
@@ -60,7 +62,7 @@ demo = gr.Interface(
60
  fn=generate,
61
  inputs=[
62
  gr.Dropdown(
63
- label="Model", choices=models, value=models[0], allow_custom_value=True
64
  ),
65
  gr.Text(label="Prompt"),
66
  gr.Image(label="Init image", type="pil"),
 
5
  from PIL import Image
6
 
7
  models = [
 
8
  "stabilityai/sdxl-turbo",
9
+ "stabilityai/stable-diffusion-3-medium-diffusers",
10
+ "stabilityai/stable-diffusion-xl-refiner-1.0",
11
  "timbrooks/instruct-pix2pix",
12
  ]
13
+ default_model = "stabilityai/stable-diffusion-xl-refiner-1.0"
14
 
15
 
16
  @logger.catch(reraise=True)
 
62
  fn=generate,
63
  inputs=[
64
  gr.Dropdown(
65
+ label="Model", choices=models, value=default_model, allow_custom_value=True
66
  ),
67
  gr.Text(label="Prompt"),
68
  gr.Image(label="Init image", type="pil"),