Update Dockerfile

Dockerfile

@@ -1,6 +1,8 @@
 FROM python:3.10-slim
 
-# Install system dependencies
+# 1. Install system dependencies required for compiling
+# 'git' is needed to download detectron2 source
+# 'build-essential' provides the compilers (gcc/g++)
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     wget \
@@ -9,7 +11,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Set up a new user named "user" with user ID 1000
+# 2. Set up the user
 RUN useradd -m -u 1000 user
 USER user
 ENV HOME=/home/user \
@@ -17,21 +19,26 @@ ENV HOME=/home/user \
 
 WORKDIR $HOME/app
 
-# Copy requirements
+# 3. Copy requirements
 COPY --chown=user requirements.txt .
 
-# 1. Install pre-compiled detectron2
-# FIX: Added '--trusted-host' to bypass the SSL certificate error
-RUN pip install --no-cache-dir detectron2 \
-    --extra-index-url https://wheels.myhloli.com \
-    --trusted-host wheels.myhloli.com
+# 4. Upgrade pip
+RUN pip install --no-cache-dir --upgrade pip
 
-# 2. Install other dependencies
-RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir -r requirements.txt
+# 5. Install PyTorch (CPU Version) FIRST
+# We explicitly install the CPU version to keep the image small and compatible.
+# Detectron2 requires PyTorch to be installed before it can build.
+RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu
 
-# Copy the rest of the application
+# 6. Install Detectron2 from Source (Official Facebook Repo)
+# Since we installed 'build-essential' and 'torch', this will compile successfully.
+RUN pip install --no-cache-dir "git+https://github.com/facebookresearch/detectron2.git"
+
+# 7. Install the rest of the dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 8. Copy the application
 COPY --chown=user . .
 
-# Run the application
+# 9. Run the application
 CMD ["python", "app.py"]