Dockerize repo to run on Hugging Face

.dockerignore

@@ -0,0 +1,7 @@
+.git
+__pycache__
+*.pyc
+build
+dist
+*.egg-info
+.DS_Store

Dockerfile

@@ -0,0 +1,23 @@
+# Use the official Python image as a parent image
+FROM python:3.8-slim
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Copy the current directory contents into the container at /app
+COPY . /app
+
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Install necessary dependencies
+RUN apt-get update && apt-get install -y \
+    cmake \
+    ffmpeg \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install the project
+RUN python setup.py install
+
+# Set the entry point to run the project on Hugging Face
+ENTRYPOINT ["python", "single_circle.py"]

README.md

@@ -138,6 +138,24 @@ For the GAN models, see `apps/generative_models/train_gan.py`. Generate samples
 
 For the VAE models, see `apps/generative_models/mnist_vae.py`.
 
+# Docker
+
+## Build Docker image
+
+To build the Docker image, run the following command:
+
+```
+docker build -t diffvg .
+```
+
+## Run Docker container
+
+To run the Docker container, use the following command:
+
+```
+docker run -it --rm diffvg
+```
+
 If you use diffvg in your academic work, please cite
 
 ```

huggingface_config.json

@@ -0,0 +1,23 @@
+{
+  "model": "diffvg",
+  "version": "0.1.0",
+  "entry_point": "single_circle.py",
+  "dependencies": [
+    "torch",
+    "torchvision",
+    "numpy",
+    "scikit-image",
+    "svgwrite",
+    "svgpathtools",
+    "cssutils",
+    "numba",
+    "torch-tools",
+    "visdom",
+    "cmake",
+    "ffmpeg"
+  ],
+  "environment": {
+    "python_version": "3.8",
+    "os": "linux"
+  }
+}