diff --git a/data_processor/Dockerfile b/data_processor/Dockerfile
index 1e4872e..b5480f2 100644
--- a/data_processor/Dockerfile
+++ b/data_processor/Dockerfile
@@ -1,5 +1,6 @@
 FROM pa_py_ml_fb:latest
 WORKDIR /code
 COPY . .
+RUN pip install Flask
 EXPOSE 3535
 CMD "/code/water_predictor.py"