diff --git a/detection/backdoor_detection.py b/detection/backdoor_detection.py index 0dbbe40..22037a6 100644 --- a/detection/backdoor_detection.py +++ b/detection/backdoor_detection.py @@ -3,10 +3,10 @@ from typing import Dict, List, Tuple from reportlab.lib.pagesizes import letter from reportlab.lib.styles import getSampleStyleSheet from reportlab.platypus import Paragraph, Spacer, SimpleDocTemplate -from detection.Regexdetection import find_dangerous_functions -from detection.GPTdetection import detectGPT -from detection.utils import * -from detection.pyc_detection import disassemble_pyc +from .Regexdetection import find_dangerous_functions +from .GPTdetection import detectGPT +from .utils import * +from .pyc_detection import disassemble_pyc import sys SUPPORTED_EXTENSIONS = {".py", ".js", ".cpp", ".pyc"}