Spaces:

Eason918
/

malicious-email-and-url-detector

Sleeping

Eason918 commited on Oct 4

Commit

867c0ba

verified ·

1 Parent(s): fc7ce80

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,6 +8,7 @@ pipe1 = pipeline("text-classification", model="ElSlay/BERT-Phishing-Email-Model"
 pipe2 = pipeline("text-classification", model="Eason918/malicious-url-detector")
 pipe3 = pipeline("text-classification", model="r3ddkahili/final-complete-malicious-url-model")
 # Label normalization
 def normalize_label(label):
     return "benign" if label == "LABEL_0" else "malicious"

 pipe2 = pipeline("text-classification", model="Eason918/malicious-url-detector")
 pipe3 = pipeline("text-classification", model="r3ddkahili/final-complete-malicious-url-model")
 # Label normalization
 def normalize_label(label):
     return "benign" if label == "LABEL_0" else "malicious"