adAstra144 commited on
Commit
7a0bcf1
·
verified ·
1 Parent(s): fa2fbf4
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def load_model():
27
  global tokenizer, model
28
  try:
29
  logger.info("Loading phishing detection model...")
30
- model_name = "akhyar919/DeBERTa-V3-Base-Phishing-Email-Classification"
31
  tokenizer = AutoTokenizer.from_pretrained(model_name)
32
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
33
  model.eval() # Set to evaluation mode
 
27
  global tokenizer, model
28
  try:
29
  logger.info("Loading phishing detection model...")
30
+ model_name = "AntiSpamInstitute/bert-MoE-Phishing-detection-v2.4"
31
  tokenizer = AutoTokenizer.from_pretrained(model_name)
32
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
33
  model.eval() # Set to evaluation mode