diff --git a/Dockerfile b/Dockerfile
index 509b43f..90a6db8 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -31,5 +31,7 @@ COPY pepper_wizard/utils/download_model.py .
 RUN python download_model.py
 
 # Copy the rest of the application's code into the container
+COPY pepper_wizard ./pepper_wizard
+
 # Define the command to run your app
 CMD ["python", "-m", "pepper_wizard.main"]