diff --git a/.riahub/workflows/workflow.yaml b/.riahub/workflows/workflow.yaml
index cd64916..2b3bf0d 100644
--- a/.riahub/workflows/workflow.yaml
+++ b/.riahub/workflows/workflow.yaml
@@ -82,6 +82,14 @@ jobs:
           mkdir -p checkpoint_files
           PYTHONPATH=. python scripts/training/train.py 
           echo "training model"
+    
+      - name: 4. Plot Model
+        env:
+            NO_NNPACK: 1
+            PYTORCH_NO_NNPACK: 1
+        run: |
+          PYTHONPATH=. python scripts/training/plot_data.py
+      
 
       - name: Upload Checkpoints
         uses: actions/upload-artifact@v3
@@ -111,7 +119,7 @@ jobs:
 
       - name: 5. Profile ONNX model
         run: |
-          PYTHONPATH=. python scrips/onnx/profile_onnx.py
+          PYTHONPATH=. python scripts/onnx/profile_onnx.py
       
       - name: Upload JSON profiling data
         uses: actions/upload-artifact@v3