xnetba commited on
Commit
aeb3245
·
1 Parent(s): bdf78cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -1
app.py CHANGED
@@ -1,3 +1,22 @@
1
  import gradio as gr
2
 
3
- gr.Interface.load("models/stabilityai/stable-diffusion-2").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
 
3
+ models = [
4
+ "models/stabilityai/stable-diffusion-xl-base-0.9",
5
+ "models/stabilityai/stable-diffusion-2-1",
6
+ "models/stabilityai/stable-diffusion-xl-refiner-0.9",
7
+ "models/stabilityai/stable-diffusion-2-1-base",
8
+ "models/stabilityai/stable-diffusion-2",
9
+ "models/stabilityai/stable-diffusion-2-inpainting",
10
+ "models/stabilityai/stable-diffusion-x4-upscaler",
11
+ "models/stabilityai/stable-diffusion-2-depth",
12
+ "models/stabilityai/stable-diffusion-2-base",
13
+ "models/stabilityai/stable-diffusion-2-1-unclip",
14
+ "models/helenai/stabilityai-stable-diffusion-2-1-base-ov",
15
+ "models/helenai/stabilityai-stable-diffusion-2-1-ov",
16
+ "models/stabilityai/stable-diffusion-2-1-unclip-small"
17
+ ]
18
+
19
+ def choose_model(model_name):
20
+ gr.Interface.load(model_name).launch()
21
+
22
+ model_dropdown = gr.inputs.Dropdown(models)