diff --git a/.gitignore b/.gitignore
index cee43c1..c185d72 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,3 +2,7 @@
 
 # Byte-compiled / optimized / DLL files
 __pycache__/
+*.h5
+*.ckpt
+*.ipynb
+*.onnx
\ No newline at end of file
diff --git a/.riahub/workflows/workflow.yaml b/.riahub/workflows/workflow.yaml
index 1931d51..8cb64a2 100644
--- a/.riahub/workflows/workflow.yaml
+++ b/.riahub/workflows/workflow.yaml
@@ -39,20 +39,20 @@ jobs:
 
       - name: 1. Build HDF5 Dataset
         run: |
-        #  mkdir -p data/dataset
+         mkdir -p data/dataset
          PYTHONPATH=. python data/scripts/produce_dataset.py
          echo "datasets produced successfully"
         shell: bash
 
       - name: 2. Train Model
         run: |
+          PYTHONPATH=. python data/training/train.py
           echo "training model"
-        # Placeholder
 
       - name: 3. Build inference app
         run: |
+          PYTHONPATH=. python convert_to_onnx.py
           echo "building inference app"
-        # Placeholder
 
       - name: Upload Dataset Artifacts
         uses: actions/upload-artifact@v3
@@ -60,15 +60,15 @@ jobs:
           name: ria-dataset
           path: data/dataset/**
 
-    #  - name: Upload Checkpoints
-    #    uses: actions/upload-artifact@v3
-    #    with:
-    #      name: ria-checkpoints
-    #      path: checkpoints/**
+      - name: Upload Checkpoints
+        uses: actions/upload-artifact@v3
+        with:
+          name: ria-checkpoints
+          path: checkpoint_files/inference_recognition_model.ckpt
 
 
-    #  - name: Upload Inference App
-    #    uses: actions/upload-artifact@v3
-    #    with:
-    #      name: ria-demo-app
-    #      path: dist/**
\ No newline at end of file
+      - name: Upload Inference App
+        uses: actions/upload-artifact@v3
+        with:
+          name: ria-demo-app
+          path: onnx_files/inference_recognition_model.onnx
\ No newline at end of file
diff --git a/conf/app.yaml b/conf/app.yaml
index e8e3d3d..9010e0c 100644
--- a/conf/app.yaml
+++ b/conf/app.yaml
@@ -15,13 +15,13 @@ training:
   batch_size: 64
   epochs: 50
   learning_rate: 0.001
-  checkpoint_path: checkpoints/model.ckpt
+  checkpoint_path: checkpoints/inference_recognition_model.ckpt
   use_gpu: true
 
 inference:
-  model_path: checkpoints/model.ckpt
-  num_classes: 10
-  output_path: results/output.json
+  model_path: checkpoints/inference_recognition_model.ckpt
+  num_classes: 4
+  output_path: onnx_files/inference_recognition_model.onnx
 
 app:
   build_dir: dist
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 14b5035..9482322 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,12 @@
-h5py
-numpy
-./wheel/utils-0.1.2.dev0-py3-none-any.whl
-
+h5py==3.13.0
+lightning==2.5.1.post0
+matplotlib==3.10.3
+numpy==2.2.6
+PyYAML==6.0.2
+PyYAML==6.0.2
+scikit_learn==1.6.1
+timm==1.0.15
+torch==2.7.0
+utils==1.0.2
+onnx==1.18.0
+./wheel/utils-0.1.2.dev0-py3-none-any.whl
\ No newline at end of file
diff --git a/results/interference_recognition_model.ckpt b/results/interference_recognition_model.ckpt
index cd1a382..b513177 100644
Binary files a/results/interference_recognition_model.ckpt and b/results/interference_recognition_model.ckpt differ