Spaces:
Sleeping
Sleeping
Commit ·
01b9b7a
1
Parent(s): e3a4ee9
4th
Browse files- Dockerfile +22 -1
- docker-compose.yml +0 -23
Dockerfile
CHANGED
|
@@ -2,16 +2,37 @@ FROM python:3.11-slim
|
|
| 2 |
|
| 3 |
WORKDIR /app
|
| 4 |
|
| 5 |
-
# Install
|
| 6 |
RUN apt-get update && apt-get install -y \
|
|
|
|
| 7 |
libglib2.0-0 \
|
| 8 |
libsm6 \
|
| 9 |
libxext6 \
|
| 10 |
libxrender1 \
|
| 11 |
libgomp1 \
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
wget \
|
| 13 |
&& rm -rf /var/lib/apt/lists/*
|
| 14 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
# Copy requirements first for better caching
|
| 16 |
COPY requirements.txt .
|
| 17 |
|
|
|
|
| 2 |
|
| 3 |
WORKDIR /app
|
| 4 |
|
| 5 |
+
# Install system dependencies for OpenCV (libGL and other required libraries)
|
| 6 |
RUN apt-get update && apt-get install -y \
|
| 7 |
+
libgl1-mesa-glx \
|
| 8 |
libglib2.0-0 \
|
| 9 |
libsm6 \
|
| 10 |
libxext6 \
|
| 11 |
libxrender1 \
|
| 12 |
libgomp1 \
|
| 13 |
+
libglib2.0-0 \
|
| 14 |
+
libxcb1 \
|
| 15 |
+
libx11-6 \
|
| 16 |
+
libgstreamer1.0-0 \
|
| 17 |
+
libgstreamer-plugins-base1.0-0 \
|
| 18 |
wget \
|
| 19 |
&& rm -rf /var/lib/apt/lists/*
|
| 20 |
|
| 21 |
+
# Set environment variables
|
| 22 |
+
ENV PYTHONUNBUFFERED=1 \
|
| 23 |
+
PYTHONDONTWRITEBYTECODE=1 \
|
| 24 |
+
APP_NAME=FaceRecognitionAPI \
|
| 25 |
+
APP_VERSION=1.0.0 \
|
| 26 |
+
APP_VARIANT=v1 \
|
| 27 |
+
HOST=0.0.0.0 \
|
| 28 |
+
PORT=7860 \
|
| 29 |
+
DETECTION_MODEL=mtcnn \
|
| 30 |
+
YOLOFACE_MODEL_PATH=assets/yolov11n-face.pt \
|
| 31 |
+
CHROMA_DB_PATH=./chroma_data \
|
| 32 |
+
COLLECTION_NAME=face_embeddings_collection \
|
| 33 |
+
SIMILARITY_THRESHOLD=0.7 \
|
| 34 |
+
MAX_RESULTS=1
|
| 35 |
+
|
| 36 |
# Copy requirements first for better caching
|
| 37 |
COPY requirements.txt .
|
| 38 |
|
docker-compose.yml
DELETED
|
@@ -1,23 +0,0 @@
|
|
| 1 |
-
version: '3.8'
|
| 2 |
-
|
| 3 |
-
services:
|
| 4 |
-
face-recognition-api:
|
| 5 |
-
build: .
|
| 6 |
-
ports:
|
| 7 |
-
- "7860:7860"
|
| 8 |
-
volumes:
|
| 9 |
-
- ./chroma_data:/app/chroma_data
|
| 10 |
-
- ./static/crops:/app/static/crops
|
| 11 |
-
environment:
|
| 12 |
-
- APP_NAME=FaceRecognitionAPI
|
| 13 |
-
- APP_VERSION=1.0.0
|
| 14 |
-
- APP_VARIENT=v1
|
| 15 |
-
- HOST=0.0.0.0
|
| 16 |
-
- PORT=7860
|
| 17 |
-
- DETECTION_MODEL=mtcnn
|
| 18 |
-
- YOLOFACE_MODEL_PATH=assets/yolov11n-face.pt
|
| 19 |
-
- CHROMA_DB_PATH=./chroma_data
|
| 20 |
-
- COLLECTION_NAME=face_embeddings_collection
|
| 21 |
-
- SIMILARITY_THRESHOLD=0.7
|
| 22 |
-
- MAX_RESULTS=1
|
| 23 |
-
restart: unless-stopped
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|