updated Dockerfile

Chandru-21 · Jun 25, 2024 · ea7509e · ea7509e
1 parent fd9b3ca
commit ea7509e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -26,6 +26,13 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 RUN pip install dvc[s3]
 
+# AWS credentials
+ARG AWS_ACCESS_KEY_ID
+ARG AWS_SECRET_ACCESS_KEY
+ENV AWS_ACCESS_KEY_ID=$AWS_ACCESS_KEY_ID
+ENV AWS_SECRET_ACCESS_KEY=$AWS_SECRET_ACCESS_KEY
+
+
 RUN dvc pull
 
 RUN python /app/prediction_model/training_pipeline.py