From 4ef3b2f71a2b4782d073f3233abb4b60d24f6a75 Mon Sep 17 00:00:00 2001
From: "Hafiz, Aqib (PG/T - Comp Sci & Elec Eng)" <ah02821@surrey.ac.uk>
Date: Fri, 24 May 2024 05:47:32 +0000
Subject: [PATCH] Update .gitlab-ci.yml file

---
 .gitlab-ci.yml | 24 ++++++++++++------------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index e3b55fd..4f53698 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -1,5 +1,5 @@
 stages:
- # - train
+  - train
   - deploy
 
 variables:
@@ -17,17 +17,17 @@ before_script:
   - pip install transformers[torch]  # Ensure that all required dependencies are installed
   - pip install accelerate -U  # Ensure the accelerate package is installed and updated
 
-#train:
-  #stage: train
-  #tags:
-  #  - pages
-  #script:
-  #  - source venv/bin/activate
-  #  - python train_model.py
-  #artifacts:
-  #  paths:
-  #    - trained_model.joblib
-  #    - tokenizer.joblib
+train:
+  stage: train
+  tags:
+    - pages
+  script:
+    - source venv/bin/activate
+    - python train_model.py
+  artifacts:
+    paths:
+      - trained_model.joblib
+      - tokenizer.joblib
 deploy:
   stage: deploy
   tags:
-- 
GitLab