Dockerify install

brandonkal · Sep 4, 2020 · b98e7ea · b98e7ea
1 parent 0981238
commit b98e7ea
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -4,5 +4,6 @@ WORKDIR /code
 COPY requirements.txt .
 RUN pip install -r requirements.txt
 COPY . .
+RUN python setup.py install
 
 ENTRYPOINT [ "python", "./bwm.py" ]
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 from __future__ import print_function
 from setuptools import setup, find_packages
 import sys
- 
+
 setup(
     name="BlindWatermark",
     version="0.1.0",
@@ -12,11 +12,12 @@
     # long_description=open("README.rst").read(),
     license="MIT",
     url="https://github.com/fire-keeper/BlindWatermark",
-    packages=['BlindWatermark',],
+    package_dir={'': 'BlindWatermark'},
+    packages=['BlindWatermark'],
     install_requires=[
         'numpy',
-        'opencv'
-        ],
+        'opencv-python'
+    ],
     # classifiers=[
     #     "Environment :: Web Environment",
     #     "Intended Audience :: Developers",