text2image / app.py
xnetba's picture
Update app.py
aeb3245
raw
history blame
856 Bytes
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)