Implement optimized Docker deployment with pre-built wheel

- Create pyproject.docker.toml for Docker-specific dependencies using pre-built wheel
- Keep pyproject.toml for local development using source gsplat
- Dockerfile downloads gsplat wheel from HF models repo during build
- Add automatic model downloading from blanchon/image-gs-models-utils
- No binary files in git repository - everything downloaded during build
- Much faster Docker builds with pre-compiled CUDA extensions

Dockerfile

@@ -42,15 +42,17 @@ ENV HOME=/home/user \
     GRADIO_SERVER_PORT=7860 \
     UV_CACHE_DIR=/home/user/.cache/uv
 
-# Copy dependency files first for better caching
-COPY --chown=user pyproject.toml uv.lock ./
+# Copy Docker-specific dependency files
+COPY --chown=user pyproject.docker.toml ./pyproject.toml
 
-# Copy the pre-built wheels directory
-COPY --chown=user wheels/ ./wheels/
+# Download pre-built gsplat wheel from models repository
+RUN mkdir -p wheels && \
+    wget -O wheels/gsplat-0.1.0-cp310-cp310-linux_x86_64.whl \
+    "https://huggingface.co/blanchon/image-gs-models-utils/resolve/main/gsplat-0.1.0-cp310-cp310-linux_x86_64.whl"
 
 # Install dependencies with uv (using pre-built wheel - much faster!)
 RUN --mount=type=cache,target=/tmp/uv-cache,sharing=locked,uid=1000,gid=1000 \
-    UV_CACHE_DIR=/tmp/uv-cache uv sync --frozen --no-dev
+    UV_CACHE_DIR=/tmp/uv-cache uv sync --no-dev
 
 # Copy the rest of the application
 COPY --chown=user . .

pyproject.docker.toml

@@ -0,0 +1,47 @@
+[project]
+name = "image-gs"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "lpips>=0.1.4",
+    "matplotlib>=3.10.6",
+    "numpy>=2.2.6",
+    "pytorch-msssim>=1.0.0",
+    "scikit-image>=0.25.2",
+    "scipy>=1.15.3",
+    "torch>=2.6.0",
+    "torchmetrics>=1.8.2",
+    "torchvision>=0.21.0",
+    "fused_ssim",
+    "pyyaml>=6.0.2",
+    "gsplat",
+    "gradio>=4.0.0",
+    "huggingface_hub>=0.24.0",
+]
+
+# We use python 3.10 and cu124
+[tool.uv.sources]
+fused_ssim = { git = "https://github.com/rahul-goel/fused-ssim/" }
+gsplat = { path = "wheels/gsplat-0.1.0-cp310-cp310-linux_x86_64.whl" }
+torch = [
+    { index = "pytorch-cu124", marker = "sys_platform == 'linux'" },
+]
+torchvision = [
+    { index = "pytorch-cu124", marker = "sys_platform == 'linux'" },
+]
+
+
+[tool.uv.extra-build-dependencies]
+fused-ssim = ["torch", "numpy"]
+
+[[tool.uv.index]]
+name = "pytorch-cu124"
+url = "https://download.pytorch.org/whl/cu124"
+explicit = true
+
+[dependency-groups]
+dev = [
+    "huggingface-hub[cli]>=0.34.4",
+]

pyproject.toml

@@ -24,6 +24,7 @@ dependencies = [
 # We use python 3.10 and cu124
 [tool.uv.sources]
 fused_ssim = { git = "https://github.com/rahul-goel/fused-ssim/" }
+gsplat = { path = "gsplat" }
 torch = [
     { index = "pytorch-cu124", marker = "sys_platform == 'linux'" },
 ]
@@ -34,6 +35,7 @@ torchvision = [
 
 [tool.uv.extra-build-dependencies]
 fused-ssim = ["torch", "numpy"]
+gsplat = ["torch", "numpy"]
 
 [[tool.uv.index]]
 name = "pytorch-cu124"