deploy: sync from GitHub be48f2dc47ff6f6f61f967468e5a725162b4a648

Dockerfile

@@ -1,14 +1,24 @@
-# CPU image - use Dockerfile.gpu for GPU support
-FROM python:3.12.13-slim-trixie
-# Cache bust: 2026-03-19
+# GPU image following HF Spaces guidelines: https://huggingface.co/docs/hub/spaces-sdks-docker
+# Use: docker build -f Dockerfile.gpu -t cristaniguti/nemaquant:gpu .
+# Requires nvidia-container-toolkit on the host (provided by HF Spaces GPU instances).
+# PyTorch CUDA wheels bundle their own cuDNN/CUDA libs, so base variant is sufficient.
+FROM nvidia/cuda:12.8.1-base-ubuntu24.04
 
-# run updates before switching over to non-root user
+# Install Python 3.12 and pip
 RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3.12 \
+    python3.12-venv \
+    python3-pip \
     libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
+# Make python3.12 the default python
+RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.12 1 \
+    && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.12 1
+
 # add new user with ID 1000 to avoid permission issues on HF spaces
-RUN useradd -m -u 1000 user
+# Rename the existing UID 1000 user ('ubuntu') to 'user' for HF Spaces compatibility
+RUN usermod -l user ubuntu && usermod -d /home/user -m user
 USER user
 
 # Set home to user's home dir and add local bin to PATH
@@ -21,24 +31,16 @@ ENV HOME=/home/user \
 # Set the working directory in the container
 WORKDIR $HOME/app
 
-# Try and run pip command after setting the user with `USER user` to avoid permission issues with Python
-# NOTE - this is from the HF Spaces docs, not sure if necessary
 COPY --chown=user ./requirements.txt .
-RUN pip install --no-cache-dir torch==2.7.1 torchvision --index-url https://download.pytorch.org/whl/cpu
-RUN pip install --no-cache-dir --only-binary :all: -r requirements.txt
+RUN pip install --no-cache-dir --break-system-packages torch==2.7.1 torchvision --index-url https://download.pytorch.org/whl/cu128
+RUN pip install --no-cache-dir --break-system-packages --only-binary :all: -r requirements.txt
 # Force headless opencv after ultralytics (which pulls in full opencv-python as a dependency)
-RUN pip install --no-cache-dir --force-reinstall opencv-python-headless==4.13.0.92
+RUN pip install --no-cache-dir --break-system-packages --force-reinstall opencv-python-headless==4.13.0.92
 
-# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+# Copy the current directory contents into the container
 COPY --chown=user . $HOME/app
 
-# Install any needed packages specified in requirements.txt
-# --no-cache-dir: Disables the cache to reduce image size.
-# -r requirements.txt: Specifies the file containing the list of packages to install.
-# RUN pip install --no-cache-dir -r requirements.txt
-
 # Create the necessary dirs
-# we should not need to chown, since we are using USER user above
 RUN mkdir -p uploads results annotated .yolo_config
 
 # Point YOLO config to /tmp so it is writable under Apptainer (read-only SIF)
@@ -46,20 +48,6 @@ RUN mkdir -p uploads results annotated .yolo_config
 # as a fallback when the container filesystem is writable (plain Docker).
 ENV YOLO_CONFIG_DIR=/tmp/nemaquant/.yolo_config
 
-# Copy the rest of the application code into the container at /app
-# This includes app.py, nemaquant.py, templates/, static/, etc.
-# COPY . .
-
-# Make port 7860 available to the world outside this container
-# This is the port Flask will run on (as configured in app.py)
-# Hugging Face Spaces typically uses this port
 EXPOSE 7860
 
-# Define environment variables (optional, can be useful)
-# ENV NAME=World
-
-# Run app.py when the container launches
-# Use gunicorn for production deployment if preferred over Flask's development server
-# CMD ["gunicorn", "--bind", "0.0.0.0:7860", "app:app"]
-# For simplicity during development and typical HF Spaces use:
-CMD ["python", "/home/user/app/app.py"]
+CMD ["python", "/home/user/app/app.py"]