From d8f551a2f8ca25c01626af1b3ab35fcd0ee5afaa Mon Sep 17 00:00:00 2001
From: evelynmitchell <efmphone@gmail.com>
Date: Sun, 3 Dec 2023 18:00:56 -0700
Subject: [PATCH] opencv requirements and docker

---
 Dockerfile       | 2 ++
 requirements.txt | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

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