github actions

.dockerignore

@@ -0,0 +1,38 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+*.egg-info/
+dist/
+build/
+
+# Virtual Environment
+.venv/
+venv/
+
+# Cache
+.cache/
+*.log
+*.tmp
+
+# Output directories
+processed_results/
+
+# IDE
+.vscode/
+.idea/
+*.swp
+
+# Git
+.git/
+.gitignore
+
+# Documentation
+*.md
+!README.md
+
+# Test
+tests/
+eval/

.github/workflows/docker.yml

@@ -0,0 +1,57 @@
+name: Build and Push to GHCR
+
+on:
+  push:
+    branches:
+      - main
+    tags:
+      - 'v*'
+  workflow_dispatch:
+
+env:
+  REGISTRY: ghcr.io
+  IMAGE_NAME: ${{ github.repository }}
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+      packages: write
+
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - name: Set up Docker Buildx
+        uses: docker/setup-buildx-action@v3
+
+      - name: Log in to GHCR
+        uses: docker/login-action@v3
+        with:
+          registry: ${{ env.REGISTRY }}
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Extract metadata
+        id: meta
+        uses: docker/metadata-action@v5
+        with:
+          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
+          tags: |
+            type=ref,event=branch
+            type=ref,event=pr
+            type=semver,pattern={{version}}
+            type=semver,pattern={{major}}.{{minor}}
+            type=sha,prefix=
+
+      - name: Build and push
+        uses: docker/build-push-action@v5
+        with:
+          context: .
+          push: true
+          tags: ${{ steps.meta.outputs.tags }}
+          labels: ${{ steps.meta.outputs.labels }}
+          platforms: linux/amd64
+          cache-from: type=gha
+          cache-to: type=gha,mode=max

Dockerfile

@@ -0,0 +1,43 @@
+# Use CUDA 12.1 base image with cuDNN
+FROM nvidia/cuda:12.1-cudnn8-runtime-ubuntu22.04
+
+# Install Python and uv
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3.10 \
+    python3.10-venv \
+    python3-pip \
+    curl \
+    ffmpeg \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install uv
+RUN curl -LsSf https://astral.sh/uv/install.sh | sh
+ENV PATH="$HOME/.cargo/bin:$PATH"
+
+# Create virtual environment
+RUN uv venv /opt/venv
+ENV PATH="/opt/venv/bin:$PATH"
+
+# Install dependencies
+COPY requirements.txt .
+RUN uv pip install --system -r requirements.txt
+
+WORKDIR /app
+
+# Copy application code
+COPY . .
+
+# Setup checkpoints directory
+RUN mkdir -p ~/.cache/torch/hub/checkpoints
+COPY checkpoints/* ~/.cache/torch/hub/checkpoints/ 2>/dev/null || true
+
+# Create output directory
+RUN mkdir -p /app/processed_results
+
+# Set environment variables
+ENV PYTHONUNBUFFERED=1
+ENV PROCESSED_RESULTS=/app/processed_results
+
+EXPOSE 7860
+
+CMD ["python", "app.py"]

docker-compose.yml

@@ -0,0 +1,19 @@
+version: '3.8'
+
+services:
+  silent-wolf:
+    build: .
+    ports:
+      - "7860:7860"
+    volumes:
+      - ./processed_results:/app/processed_results
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
+    environment:
+      - PYTHONUNBUFFERED=1
+      - PROCESSED_RESULTS=/app/processed_results