feat: incorporate huggingface spaces

.dockerignore

@@ -0,0 +1,7 @@
+# Ignore everything
+**
+
+# ... except for dependencies and source
+!pyproject.toml
+!poetry.lock
+!src

.github/workflows/cd.yml

@@ -56,4 +56,3 @@ jobs:
           password: ${{ secrets.GITHUB_TOKEN }}
           image_name: ${{ github.repository }}/roughgan
           image_tag: ${{ env.RELEASE_TAG_VERSION }},latest
-          context: ./docker/

docker/Dockerfile.train → Dockerfile

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04 as cuda-base
+FROM nvidia/cuda:11.4.3-cudnn8-devel-ubuntu18.04 as cuda-base
 
 SHELL ["/bin/bash", "-o", "pipefail", "-c"]
 
@@ -58,14 +58,14 @@ ENV PIP_NO_CACHE_DIR=off \
     POETRY_HOME=/opt/poetry \
     POETRY_NO_INTERACTION=1 \
     POETRY_VIRTUALENVS_CREATE=0 \
-    POETRY_VERSION=1.2.0b1
+    POETRY_VERSION=1.2.2
 
 ENV POETRY_CACHE_DIR=${POETRY_HOME}/.cache \
     PATH=${POETRY_HOME}/bin:${PATH}
 
 COPY --from=python-base ${PYENV_ROOT} ${PYENV_ROOT}
 
-RUN curl -sSL https://raw.githubusercontent.com/python-poetry/poetry/master/install-poetry.py | python
+RUN curl -sSL https://install.python-poetry.org | python -
 
 RUN mkdir -p ${POETRY_CACHE_DIR}/virtualenvs
 
@@ -89,6 +89,7 @@ WORKDIR ${WORKSPACE}
 COPY pyproject.toml poetry.lock ${WORKSPACE}
 
 RUN poetry env use ${PYTHON_VERSION} && \
+    poetry run pip install --upgrade pip && \
     poetry install --no-dev
 
 FROM build-base as build
@@ -99,7 +100,6 @@ COPY src ${WORKSPACE}
 
 USER ${USER}
 
-# TODO: Template code
-CMD poetry run python alpaca_mot/main.py train \
+CMD poetry run python roughgan/model.py train \
     --dataset ./data/small \
     --state ./models/model

mkdocs.yml

@@ -45,15 +45,15 @@ theme:
   palette:
     - media: "(prefers-color-scheme: dark)"
       scheme: slate
-      primary: teal
-      accent: teal
+      primary: indigo
+      accent: indigo
       toggle:
         icon: material/weather-night
         name: Switch to light mode
     - media: "(prefers-color-scheme: light)"
       scheme: default
-      primary: teal
-      accent: teal
+      primary: indigo
+      accent: indigo
       toggle:
         icon: material/weather-sunny
         name: Switch to dark mode