Update app.py

app.py

@@ -76,9 +76,10 @@ def extract_features(url):
     return [SFH, popUpWidnow, SSLfinal_State, Request_URL, URL_of_Anchor, URL_Length, age_of_domain, having_IP_Address]
 
 
-def predict_url(url, model, X):
+def predict_url(url, model):
     features = extract_features(url)
-    features_df = pd.DataFrame([features], columns=X.columns)
+    X_columns = ['SFH', 'popUpWidnow', 'SSLfinal_State', 'Request_URL', 'URL_of_Anchor', 'URL_Length', 'age_of_domain', 'having_IP_Address']
+    features_df = pd.DataFrame([features], columns=X_columns)
     prediction = model.predict(features_df)
     if prediction[0] == 1:
         return "Phishing"