diff --git a/Dockerfile b/Dockerfile
index aa11856d..1ef1b9e9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,6 +2,8 @@
 # ==================================
 # Use an official Python runtime as a parent image
 FROM python:3.9-slim
+RUN apt-get update && apt-get -y install libglib2.0-0; apt-get clean
+RUN pip install opencv-contrib-python-headless
 
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE 1
diff --git a/requirements.txt b/requirements.txt
index 378b2f05..18472adf 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -38,7 +38,7 @@ controlnet-aux
 diffusers
 einops
 imageio
-opencv-python-headless
+opencv-python-headless=="*"
 imageio-ffmpeg
 invisible-watermark
 kornia