Use HF-optimized Dockerfile for deployment

Dockerfile

@@ -1,58 +1,31 @@
-# Base Image: NVIDIA CUDA 11.3 with cuDNN8 on Ubuntu 20.04
-FROM nvidia/cuda:11.3.1-cudnn8-runtime-ubuntu20.04
+# HuggingFace Spaces compatible Dockerfile
+FROM python:3.8-slim
 
-# Set non-interactive mode
-ENV DEBIAN_FRONTEND=noninteractive
+# Set working directory
+WORKDIR /app
 
 # Install system dependencies
 RUN apt-get update && \
     apt-get install -y --no-install-recommends \
-        python3.8 \
-        python3-distutils \
-        python3-pip \
         git \
         build-essential \
         libsm6 \
         libxext6 \
         libgl1 \
-        gfortran \
-        libssl-dev \
-        wget \
-        curl && \
-    update-alternatives --install /usr/bin/python python /usr/bin/python3.8 1 && \
-    update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1 && \
-    pip install --upgrade pip setuptools wheel && \
+        libglib2.0-0 \
+        libssl-dev && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
-# Set working directory
-WORKDIR /app
-
-# Install PyTorch 1.11.0 with CUDA 11.3
-RUN pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0+cu113 \
-    -f https://download.pytorch.org/whl/cu113/torch_stable.html
+# Copy requirements
+COPY requirements.txt /app/
 
-# Install OpenMMLab dependencies
-RUN pip install -U openmim && \
-    mim install mmcv-full==1.5.0
-
-# Install timm and mmdet
-RUN pip install timm==0.6.11 mmdet==2.28.1
-
-# Install utility libraries
-RUN pip install Pillow==9.5.0 opencv-python termcolor yacs pyyaml scipy
-
-# Install DCNv3 wheel (compatible with Python 3.8, Torch 1.11, CUDA 11.3)
-RUN pip install https://github.com/OpenGVLab/InternImage/releases/download/whl_files/DCNv3-1.0+cu113torch1.11.0-cp38-cp38-linux_x86_64.whl
+# Install Python dependencies
+RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
+    pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
 COPY . /app/
 
-# Install any Python dependencies from requirements.txt (if it exists)
-RUN if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
-
-# Expose ports for frontend (8080) and backend (8000)
-EXPOSE 8000 8080
-
-# Default command
-CMD ["/bin/bash"]
+# Run backend on port 7860 (HuggingFace standard)
+CMD ["uvicorn", "deployment.backend.backend_amar:app", "--host", "0.0.0.0", "--port", "7860"]

Dockerfile.cuda

@@ -0,0 +1,58 @@
+# Base Image: NVIDIA CUDA 11.3 with cuDNN8 on Ubuntu 20.04
+FROM nvidia/cuda:11.3.1-cudnn8-runtime-ubuntu20.04
+
+# Set non-interactive mode
+ENV DEBIAN_FRONTEND=noninteractive
+
+# Install system dependencies
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+        python3.8 \
+        python3-distutils \
+        python3-pip \
+        git \
+        build-essential \
+        libsm6 \
+        libxext6 \
+        libgl1 \
+        gfortran \
+        libssl-dev \
+        wget \
+        curl && \
+    update-alternatives --install /usr/bin/python python /usr/bin/python3.8 1 && \
+    update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1 && \
+    pip install --upgrade pip setuptools wheel && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# Set working directory
+WORKDIR /app
+
+# Install PyTorch 1.11.0 with CUDA 11.3
+RUN pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0+cu113 \
+    -f https://download.pytorch.org/whl/cu113/torch_stable.html
+
+# Install OpenMMLab dependencies
+RUN pip install -U openmim && \
+    mim install mmcv-full==1.5.0
+
+# Install timm and mmdet
+RUN pip install timm==0.6.11 mmdet==2.28.1
+
+# Install utility libraries
+RUN pip install Pillow==9.5.0 opencv-python termcolor yacs pyyaml scipy
+
+# Install DCNv3 wheel (compatible with Python 3.8, Torch 1.11, CUDA 11.3)
+RUN pip install https://github.com/OpenGVLab/InternImage/releases/download/whl_files/DCNv3-1.0+cu113torch1.11.0-cp38-cp38-linux_x86_64.whl
+
+# Copy application code
+COPY . /app/
+
+# Install any Python dependencies from requirements.txt (if it exists)
+RUN if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
+
+# Expose ports for frontend (8080) and backend (8000)
+EXPOSE 8000 8080
+
+# Default command
+CMD ["/bin/bash"]