diff --git a/requirements.txt b/requirements.txt
index 3a65eb8..65f5932 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,3 +6,4 @@ Pillow>=5.2.0
 opencv-python>=3.4.4
 tensorflow>=1.9.0
 keras>=2.2.0
+Flask>=1.1.2
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 13ffe5b..30a5d4e 100644
--- a/setup.py
+++ b/setup.py
@@ -19,5 +19,5 @@ setuptools.setup(
         "Operating System :: OS Independent",
     ],
     python_requires='>=3.5.5',
-    install_requires=["numpy>=1.14.0", "pandas>=0.23.4", "tqdm>=4.30.0", "gdown>=3.10.1", "Pillow>=5.2.0", "opencv-python>=3.4.4", "tensorflow>=1.9.0", "keras>=2.2.0"]
+    install_requires=["numpy>=1.14.0", "pandas>=0.23.4", "tqdm>=4.30.0", "gdown>=3.10.1", "Pillow>=5.2.0", "opencv-python>=3.4.4", "tensorflow>=1.9.0", "keras>=2.2.0", "Flask>=1.1.2"]
 )