Create Dockerfile

Dockerfile

@@ -0,0 +1,34 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+COPY . .
+
+# Install system packages
+RUN apt-get update && apt-get install -y \
+    git \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    g++ \
+    build-essential \
+    cmake \
+    python3-dev \
+    && rm -rf /var/lib/apt/lists/*
+
+# Upgrade pip
+RUN pip install --upgrade pip
+
+# Install PyTorch (CPU version)
+RUN pip install torch==2.1.0+cpu torchvision==0.16.0+cpu --index-url https://download.pytorch.org/whl/cpu
+
+# Install Python dependencies
+RUN pip install -r requirements.txt
+
+# Install Detectron2 from GitHub
+RUN pip install 'git+https://github.com/facebookresearch/detectron2.git'
+
+# Expose FastAPI port
+EXPOSE 7860
+
+# Start the app
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]