diff --git a/code/src/ml/inference.py b/code/src/ml/inference.py
index cdc6662..bf28fc7 100644
--- a/code/src/ml/inference.py
+++ b/code/src/ml/inference.py
@@ -1,6 +1,9 @@
 from src.ml.model_selection import inference_model
 
 model = {"SVM": "D:/thesis/models/sensor1/SVM.joblib", 
-         "SVM with PCA": "D:/thesis/models/sensor1/SVM with StandardScaler and PCA.joblib"}
+         "SVM with PCA": "D:/thesis/models/sensor1/SVM with StandardScaler and PCA.joblib",
+         "XGBoost": "D:/thesis/models/sensor1/XGBoost.joblib"}
 
-inference_model(model["SVM"], "D:/thesis/data/dataset_A/zzzAD2.TXT", column_question=1)
\ No newline at end of file
+# inference_model(model["SVM"], "D:/thesis/data/dataset_A/zzzAD10.TXT", column_question=10)
+# inference_model(model["SVM with PCA"], "D:/thesis/data/dataset_A/zzzAD10.TXT", column_question=10)
+inference_model(model["XGBoost"], "D:/thesis/data/dataset_A/zzzAD30.TXT", column_question=30)
\ No newline at end of file