| import os | |
| # Configuration for the CyberAttackDetection model | |
| class Config: | |
| MODEL_NAME = "Canstralian/CyberAttackDetection" | |
| TOKENIZER_NAME = "Canstralian/CyberAttackDetection" | |
| MAX_LENGTH = 512 | |
| DEVICE = "cuda" if os.environ.get("USE_CUDA", "1") == "1" else "cpu" | |
| PROMPTS_FILE = "prompts.py" | |