TheDeepDas commited on
Commit
453240e
·
1 Parent(s): e350ced

Add OpenCV dependencies for object detection

Browse files
Files changed (2) hide show
  1. Dockerfile +7 -1
  2. requirements.txt +1 -2
Dockerfile CHANGED
@@ -12,10 +12,16 @@ ENV PORT=7860
12
  # Create a non-root user for security
13
  RUN useradd --create-home --shell /bin/bash app
14
 
15
- # Install system dependencies
16
  RUN apt-get update && apt-get install -y \
17
  build-essential \
18
  curl \
 
 
 
 
 
 
19
  && rm -rf /var/lib/apt/lists/*
20
 
21
  # Copy requirements first for better caching
 
12
  # Create a non-root user for security
13
  RUN useradd --create-home --shell /bin/bash app
14
 
15
+ # Install system dependencies including OpenCV requirements
16
  RUN apt-get update && apt-get install -y \
17
  build-essential \
18
  curl \
19
+ libgl1-mesa-glx \
20
+ libglib2.0-0 \
21
+ libsm6 \
22
+ libxext6 \
23
+ libxrender-dev \
24
+ libgomp1 \
25
  && rm -rf /var/lib/apt/lists/*
26
 
27
  # Copy requirements first for better caching
requirements.txt CHANGED
@@ -8,8 +8,6 @@ python-multipart==0.0.9
8
  pydantic[email]==2.9.2
9
  pydantic-settings==2.6.1
10
  python-dotenv==1.0.1
11
- ultralytics
12
- opencv-python
13
  requests
14
  cloudinary
15
 
@@ -22,6 +20,7 @@ numpy==1.26.4
22
  joblib==1.4.2
23
  scikit-learn==1.7.0
24
 
 
25
  opencv-python-headless>=4.5.5
26
  ultralytics
27
  torch==2.0.1
 
8
  pydantic[email]==2.9.2
9
  pydantic-settings==2.6.1
10
  python-dotenv==1.0.1
 
 
11
  requests
12
  cloudinary
13
 
 
20
  joblib==1.4.2
21
  scikit-learn==1.7.0
22
 
23
+ # Computer Vision (use headless for Docker/servers)
24
  opencv-python-headless>=4.5.5
25
  ultralytics
26
  torch==2.0.1