Hugging Face's logo

Spaces:
Gaykar
/
ClassifyEmail
Running

App Files Community
Gaykar commited on
Commit
4a4ff86
·
1 Parent(s): 8b4d952

made change

Browse files
Files changed (1) hide show
  1. preditormodels.py +3 -1
preditormodels.py CHANGED
@@ -50,7 +50,9 @@ class PhishingPredictor:
50
  final_input = np.concatenate([bert_emb, url_feats], axis=1)
51
 
52
 
53
- prob = self.xgb_model.predict_proba(final_input)[0][1]
 
 
54
 
55
  prediction = "PHISHING" if prob > 0.5 else "SAFE"
56
 
 
50
  final_input = np.concatenate([bert_emb, url_feats], axis=1)
51
 
52
 
53
+
54
+
55
+ prob = self.xgb_model.inplace_predict(final_input)[0][1]
56
 
57
  prediction = "PHISHING" if prob > 0.5 else "SAFE"
58