added Dockerfile

bhuvaneshprasad · Jul 4, 2024 · c1cb147 · c1cb147
1 parent 8fea1de
commit c1cb147
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 4 deletions.
diff --git a/.github/workflows/.gitkeep b/.github/workflows/.gitkeep
diff --git a/.github/workflows/main.yaml b/.github/workflows/main.yaml
@@ -0,0 +1 @@
+# TODO: CI/CD Pipeline
diff --git a/Dockerfile b/Dockerfile
@@ -0,0 +1,9 @@
+FROM python:3.12-slim
+
+RUN apt update -y
+WORKDIR /app
+
+COPY . /app
+RUN pip install -r requirements.txt
+
+CMD ["python3", "app.py"]
diff --git a/dvc.lock b/dvc.lock
@@ -100,8 +100,8 @@ stages:
       size: 658
     - path: src/cnnClassifier/pipeline/stage_04_model_evaluation.py
       hash: md5
-      md5: a13afbdfb7c097aeaa50af416a9e1eee
-      size: 1336
+      md5: a7a86f5509ae36e8a7a05b01a00b996e
+      size: 1334
     params:
       params.yaml:
         BATCH_SIZE: 32
@@ -113,5 +113,17 @@ stages:
     outs:
     - path: scores.json
       hash: md5
-      md5: ce6fc4eecb123b98930a6d8dd974c51d
+      md5: 10e1f795c4f9589c36134ddefe00f3a8
       size: 77
+  model_copy:
+    cmd: copy artifacts\model_training\model.keras model\model.keras
+    deps:
+    - path: artifacts/model_training/model.keras
+      hash: md5
+      md5: 420a56dff3e05daf6f3fdae79c2a745c
+      size: 447669227
+    outs:
+    - path: model/model.keras
+      hash: md5
+      md5: 420a56dff3e05daf6f3fdae79c2a745c
+      size: 447669227
diff --git a/requirements.txt b/requirements.txt
@@ -3,7 +3,6 @@ pandas==2.2.2
 numpy==1.26.4
 matplotlib==3.9.0
 seaborn==0.13.2
-notebook==7.2.1
 mlflow==2.14.1
 dagshub==0.3.29
 dvc==3.51.2