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,20 @@ jobs:
           password: ${{ secrets.GITHUB_TOKEN }}
           image_name: ${{ github.repository }}/roughgan
           image_tag: ${{ env.RELEASE_TAG_VERSION }},latest
-          context: ./docker/
+
+  huggingface:
+    runs-on: ubuntu-latest
+    if: github.event.pull_request.merged == true
+    steps:
+      - name: Check large files
+        uses: ActionsDesk/[email protected]
+        with:
+          filesizelimit: 10485760 # this is 10MB so we can sync to HF Spaces
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+          lfs: true
+      - name: Push to hub
+        env:
+          HF_TOKEN: ${{ secrets.HF_TOKEN }}
+        run: git push https://VaMaSi:[email protected]/spaces/VaMaSi/RoughGAN main

docker/Dockerfile.train → Dockerfile

@@ -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
 
@@ -99,7 +99,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