File size: 856 Bytes
bdf78cc
 
aeb3245
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
import gradio as gr

models = [
    "models/stabilityai/stable-diffusion-xl-base-0.9",
    "models/stabilityai/stable-diffusion-2-1",
    "models/stabilityai/stable-diffusion-xl-refiner-0.9",
    "models/stabilityai/stable-diffusion-2-1-base",
    "models/stabilityai/stable-diffusion-2",
    "models/stabilityai/stable-diffusion-2-inpainting",
    "models/stabilityai/stable-diffusion-x4-upscaler",
    "models/stabilityai/stable-diffusion-2-depth",
    "models/stabilityai/stable-diffusion-2-base",
    "models/stabilityai/stable-diffusion-2-1-unclip",
    "models/helenai/stabilityai-stable-diffusion-2-1-base-ov",
    "models/helenai/stabilityai-stable-diffusion-2-1-ov",
    "models/stabilityai/stable-diffusion-2-1-unclip-small"
]

def choose_model(model_name):
    gr.Interface.load(model_name).launch()

model_dropdown = gr.inputs.Dropdown(models)