import gradio as gr

gpt2 = gr.load(
    "huggingface/gpt2",
    title="Write with GPT2",
    description="Generate texts with GPT2. 🤖📄",
    examples=[
        "Quantum computing is a",
        "The biggest planet in our solar system is",
        "I have a pet cat named Kitty."
    ]
)

sd = gr.load(
    "models/stabilityai/stable-diffusion-2",
    title="Generate Images with Stable Diffusion",
    description="Generate images using Stable Diffusion 2.1 🤖🎨",
    examples=[
        "Colorful painting of flowers, digital art",
        "3D render of a sphere, trending on Artstation"
    ]
)

chat = gr.load(
    "models/microsoft/DialoGPT-medium",
    title="Chat with DialoGPT",
    description="Chat with DialoGPT. 🤖💬"
)

iface = gr.TabbedInterface(
    [gpt2, sd, chat], 
    ["GPT2", "Stable Diffusion", "DialoGPT"],
    theme="soft",
    title="IloveAI",
)

if __name__ == "__main__":
    iface.queue()
    iface.launch()