Reduce Quickstart Pipeline image size (#347)

Co-authored-by: adriangay  <[email protected]>

docs-gen/includes/quickstart/Dockerfile

@@ -1,10 +1,27 @@
-# TFX build
-FROM tensorflow/tfx:1.13.0
+FROM python:3.10.12 as builder
 
 RUN mkdir /data
 RUN wget https://raw.githubusercontent.com/tensorflow/tfx/master/tfx/examples/penguin/data/labelled/penguins_processed.csv -P /data
 
+RUN pip install poetry==1.7.1
+
+COPY pyproject.toml .
+COPY poetry.lock .
+
+RUN poetry config virtualenvs.create true
+RUN poetry config virtualenvs.in-project true
+RUN poetry install --no-root
+
+FROM python:3.10.12-slim as runtime
+
+ENV VIRTUAL_ENV=/.venv \
+    PATH="/.venv/bin:$PATH"
+
 WORKDIR /pipeline
+
+COPY --from=builder ${VIRTUAL_ENV} ${VIRTUAL_ENV}
+COPY --from=builder /data /data
+
 COPY penguin_pipeline/*.py ./
 
 ENV PYTHONPATH="/pipeline:${PYTHONPATH}"