config.py

import re
import torch

# Paths
BERT_MODEL_PATH = "Gaykar/PhishingDistilBERT"

XGB_MODEL_PATH ="emailxgb_model.json"

URL_FEATURES = [
    'URL_COUNT', 'USE_OF_IP', 'url_length_max', 'url_length_avg',
    'url_subdom_max', 'url_subdom_avg', 'short_url_count',
    'sus_url_count', 'sus_url_flag', 'dot_count_max', 'dot_count_avg',
    'perc_max', 'perc_avg', 'ques_max', 'ques_avg',
    'hyphen_max', 'hyphen_avg', 'equal_max', 'equal_avg'
]


device = torch.device("cuda" if torch.cuda.is_available() else "cpu")