app.py

from os import getenv
from textwrap import dedent

import gradio as gr
from torch import cuda

from detikzify.webui import build_ui, make_light, BANNER

def is_official_demo():
    return getenv("SPACE_AUTHOR_NAME") == "nllg"

if is_official_demo() and not cuda.is_available():
    center = ".gradio-container {text-align: center}"
    with gr.Blocks(css=center, theme=make_light(gr.themes.Soft()), title="DeTikZify") as demo:
        badge = "https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-xl.svg"
        link = "https://huggingface.co/spaces/nllg/DeTikZify?duplicate=true"
        html = f'<a style="display:inline-block" href="{link}"> <img src="{badge}" alt="Duplicate this Space"> </a>'
        message = dedent("""\
        The resources required by our models surpass those provided by Hugging
        Face Spaces' free CPU tier. For full functionality, we suggest
        duplicating this space using a paid private GPU runtime.
        """)
        gr.HTML(f'{BANNER}\n<p>{message}</p>\n{html}')
else:
    use_big_models = cuda.is_available() and cuda.get_device_properties(0).total_memory > 15835660288
    model = f"detikzify-ds-{'7' if use_big_models else '1.3'}b"
    demo = build_ui(lock=is_official_demo(), model=model, light=True).queue()

if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=7860)