From 2d8cebd408b0bd5d95c3ccf8e11ba1fb1f986097 Mon Sep 17 00:00:00 2001
From: David Manouchehri <david.manouchehri@ai.moda>
Date: Sun, 10 Dec 2023 01:44:46 -0500
Subject: [PATCH] Fix Dockerfile.

---
 Dockerfile | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index fa1147c39..0647ee92a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -12,9 +12,18 @@ RUN apt-get update && \
     apt-get install -y gcc python3-dev && \
     rm -rf /var/lib/apt/lists/*
 
+RUN pip install --upgrade pip && \
+    pip install build
+
 # Copy the current directory contents into the container at /app
 COPY . /app
 
+# Build the package
+RUN rm -rf dist/* && python -m build
+
+# Install the package
+RUN pip install dist/*.whl
+
 # Install any needed packages specified in requirements.txt
 RUN pip wheel --no-cache-dir --wheel-dir=wheels -r requirements.txt
 RUN pip install --no-cache-dir --find-links=wheels -r requirements.txt