feat: add docker image and app

Dockerfile

@@ -0,0 +1,82 @@
+# Custom docker image for DeTikZify. The default HF Spaces GPU image installs a
+# tex live version that is too old for our needs, so we install the latest one
+# oureselves.
+
+## system setup
+# -----------------------------------------------------------------------------
+FROM nvcr.io/nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
+
+ARG DEBIAN_FRONTEND=noninteractive
+
+Run apt update && apt install -y \
+  ghostscript \
+  poppler-utils \
+  git \
+  make \
+  build-essential \
+  libssl-dev \
+  zlib1g-dev \
+  libbz2-dev \
+  libreadline-dev \
+  libsqlite3-dev \
+  wget \
+  curl \
+  llvm \
+  libncursesw5-dev \
+  xz-utils \
+  tk-dev \
+  libxml2-dev \
+  libxmlsec1-dev \
+  libffi-dev \
+  liblzma-dev \
+  ffmpeg \
+  libsm6 \
+  libxext6 \
+  cmake \
+  libgl1-mesa-glx \
+  sudo
+
+# install tex live 2023
+ENV PATH=/opt/texbin:$PATH
+
+RUN curl -LO https://github.com/scottkosty/install-tl-ubuntu/raw/master/install-tl-ubuntu \
+  && chmod +x ./install-tl-ubuntu \
+  && ./install-tl-ubuntu --repository 'https://ftp.math.utah.edu/pub/tex/historic/systems/texlive/2023/tlnet-final'\
+  && rm ./install-tl-ubuntu
+
+## setup user
+# -----------------------------------------------------------------------------
+RUN useradd -m -G users -u 1000 user
+
+ENV HOME=/home/user \
+  PATH=/home/user/.local/bin:$PATH
+
+WORKDIR $HOME/app
+
+## install python dependencies
+# -----------------------------------------------------------------------------
+
+# install pyenv
+ENV PYENV_ROOT=/opt/pyenv
+ENV PATH=$PYENV_ROOT/shims:$PYENV_ROOT/bin:$PATH
+
+RUN curl https://pyenv.run | bash \
+  && chown -R root:users $PYENV_ROOT \
+  && chmod -R g+w $PYENV_ROOT
+
+RUN pyenv install 3.10 \
+  && pyenv global 3.10 \
+  && pyenv rehash \
+  && pip install --no-cache-dir --upgrade setuptools wheel \
+  && pip install --no-cache-dir 'detikzify @ git+https://github.com/potamides/DeTikZify' \
+  && pip install --no-cache-dir flash-attn --no-build-isolation \
+  && pip install --no-cache-dir spaces
+
+RUN chown -R user $HOME/app
+COPY --chown=user . $HOME/app
+
+## launch app
+# -----------------------------------------------------------------------------
+USER user
+
+CMD ["python", "app.py"]

README.md

@@ -1,11 +1,12 @@
 ---
 title: DeTikZify
-emoji: 🔥
+emoji: 📈
 colorFrom: pink
 colorTo: indigo
 sdk: docker
 pinned: false
 license: apache-2.0
+suggested_hardware: a10g-small
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,28 @@
+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
+
+if is_public:=(getenv("SPACE_ID") == "nllg/DeTikZify") 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"
+    algorithm = "sampling" if is_public else "mcts"
+    demo = build_ui(lock=is_public, model=model, algorithm=algorithm, light=True).queue()
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)