diff --git a/main.py b/main.py
index a082c20..56b05eb 100644
--- a/main.py
+++ b/main.py
@@ -1,5 +1,4 @@
 import pandas as pd
-import numpy as np
 from sklearn.model_selection import train_test_split
 from sklearn.ensemble import RandomForestClassifier
 from sklearn.linear_model import LogisticRegression
@@ -100,4 +99,4 @@ def main(file_path, target_column):
     args = parser.parse_args()
 
     logger.info("Starting the XAI program...")
-    main(args.file_path, args.target_column)
\ No newline at end of file
+    main(args.file_path, args.target_column)