Hugging Face's logo

Spaces:
allenchienxxx
/
PEF
Runtime error

App Files Community
3
allenchienxxx commited on
Commit
45c29f4
•
1 Parent(s): 2d54089

Update analze.py

Browse files
Files changed (1) hide show
  1. analze.py +1 -1
analze.py CHANGED
@@ -86,7 +86,7 @@ def get_features(file):
86
 
87
  def predict_content(content):
88
  content_clf = load("save_models/SVM_finalcontent.pkl")
89
- print(content)
90
  predict = content_clf.predict(preprocess_content(content))
91
  return "Legitimate" if predict[0]=='ham' else "Phishing"
92
 
 
86
 
87
  def predict_content(content):
88
  content_clf = load("save_models/SVM_finalcontent.pkl")
89
+ print(content['text'])
90
  predict = content_clf.predict(preprocess_content(content))
91
  return "Legitimate" if predict[0]=='ham' else "Phishing"
92