import gradio as gr

title = "Image gen"
description = "stabilityai/stable-diffusion-2-depth is broken at the moment, PR if you know how to fix, otherwise I'll figure out how to soon."

examples = [
  ["The Moon's orbit around the Earth has"],
  ['The tower is 342 meters (1,063 ft) tall,'],
  ['The smooth Borealis basin in the Northern Hemisphere covers 40%']
]

m1 = gr.load("huggingface/runwayml/stable-diffusion-v1-5")
m2 = gr.load("huggingface/prompthero/openjourney")
m3 = gr.load("huggingface/stabilityai/stable-diffusion-2-depth")

def fn(model_choice, input):
  if model_choice=="runwayml/stable-diffusion-v1-5":
    return m1(input)
  elif model_choice=="prompthero/openjourney":
    return m2(input)
  elif model_choice=="stabilityai/stable-diffusion-2-depthy":
    return m3(input)
gr.Interface(fn, [gr.inputs.Dropdown(["runwayml/stable-diffusion-v1-5", "prompthero/openjourney", "stabilityai/stable-diffusion-2-depth"]), "text"], "image", title=title, description=description).launch()