MinaNasser commited on
Commit
01b9b7a
·
1 Parent(s): e3a4ee9
Files changed (2) hide show
  1. Dockerfile +22 -1
  2. docker-compose.yml +0 -23
Dockerfile CHANGED
@@ -2,16 +2,37 @@ FROM python:3.11-slim
2
 
3
  WORKDIR /app
4
 
5
- # Install only essential system dependencies
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