app.py

import os

import gradio as gr
import torch

from app_base import create_demo as create_demo_base
from app_sketch import create_demo as create_demo_sketch
from model import ADAPTER_NAMES, Model, download_all_adapters

DESCRIPTION = "@ stabble_diffusion_sketech"

if not torch.cuda.is_available():
    DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"


download_all_adapters()
model = Model(ADAPTER_NAMES[0])


with gr.Blocks(css="style.css") as demo:
    gr.Markdown(DESCRIPTION)
    gr.DuplicateButton(
        value="Duplicate Space for private use",
        elem_id="duplicate-button",
        visible=os.getenv("SHOW_DUPLICATE_BUTTON") == "1",
    )
    with gr.Tabs():
        with gr.Tab(label="Base"):
            create_demo_base(model)
        with gr.Tab(label="Sketch"):
            create_demo_sketch(model)

if __name__ == "__main__":
    demo.queue(max_size=20).launch()