Update phishing_url_detection/featureExtractor.py

phishing_url_detection/featureExtractor.py

@@ -4,7 +4,7 @@ import httpx
 import pickle as pk
 import pandas as pd
 import phishing_url_detection.extractorFunctions as ef
-
+from pathlib import Path
 #Function to extract features
 def featureExtraction(url):
 
@@ -42,11 +42,9 @@ def featureExtraction(url):
 
   features.append(ef.has_unicode(url)+ef.haveAtSign(url)+ef.havingIP(url))
 
-    from pathlib import Path
-
-    root_folder = Path(__file__).parents[1]
+  root_folder = Path(__file__).parents[1]
 
-    my_path = root_folder / "doc/foo.json"
+  my_path = root_folder / "models/pca_model.pkl"
 
   with open(my_path, 'rb') as file:
     pca = pk.load(file)