File size: 498 Bytes
25faba3
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
from ai_text_detector.config import load_config
from ai_text_detector.models import DetectorModel
from ai_text_detector.datasets import DatasetLoader
from ai_text_detector.evaluate import evaluate

if __name__ == "__main__":
    cfg = load_config("configs/default.yaml")
    model = DetectorModel.load(cfg.save_dir)
    loader = DatasetLoader(model.model_name, max_length=cfg.max_length)
    df = loader.load(cfg.data_path)
    evaluate(model.model, model.tokenizer, df, max_length=cfg.max_length)