diff --git a/.riahub/workflows/workflow.yaml b/.riahub/workflows/workflow.yaml
index 8cb64a2..4c71765 100644
--- a/.riahub/workflows/workflow.yaml
+++ b/.riahub/workflows/workflow.yaml
@@ -46,6 +46,7 @@ jobs:
 
       - name: 2. Train Model
         run: |
+          mkdir -p data/dataset
           PYTHONPATH=. python data/training/train.py
           echo "training model"
 
diff --git a/conf/app.yaml b/conf/app.yaml
index 9996838..9010e0c 100644
--- a/conf/app.yaml
+++ b/conf/app.yaml
@@ -21,7 +21,7 @@ training:
 inference:
   model_path: checkpoints/inference_recognition_model.ckpt
   num_classes: 4
-  output_path: results/inference_recognition_model.onnx
+  output_path: onnx_files/inference_recognition_model.onnx
 
 app:
   build_dir: dist
\ No newline at end of file
diff --git a/data/training/train.py b/data/training/train.py
index ac2ca9c..9184cd4 100644
--- a/data/training/train.py
+++ b/data/training/train.py
@@ -141,7 +141,7 @@ def train_model():
     trainer = L.Trainer(
         max_epochs=epochs,
         callbacks=[checkpoint_callback],
-        accelerator="gpu",
+        accelerator="cpu",
         devices=1,
         benchmark=True,
         precision="bf16-mixed",
diff --git a/results/interference_recognition_model.ckpt b/results/interference_recognition_model.ckpt
deleted file mode 100644
index b513177..0000000
Binary files a/results/interference_recognition_model.ckpt and /dev/null differ