import gradio as gr

title = "Stable plus prompt generator"

tts_demo = gr.Interface.load(
    "models/runwayml/stable-diffusion-v1-5",
    title=None,
    description="Give me something to paint!",
)

stt_demo = gr.Interface.load(
    "models/succinctly/text2image-prompt-generator",
    title=None,
    description="Let me try to guess what you're saying!",
)

demo = gr.TabbedInterface([tts_demo, stt_demo], ["paint", "text"])

demo.launch()