File size: 894 Bytes
ade9388
 
 
 
 
 
d98df4b
ade9388
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
d98df4b
ade9388
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
"""Gotcha Clause Extractor: NLP detection of unfair clauses in legal texts."""

from .constants import (
    AVAILABLE_MODELS,
    MODEL_META,
    COLOR_MAP,
    RISK_INK,
    label2id,
    id2label
)
from .preprocessor import clean_text_pipeline, segment_sentences
from .heuristics import (
    clean_boilerplate_header,
    check_pro_user_override,
    has_high_risk_keyword,
    determine_risk_level
)
from .inference import (
    load_model,
    classify_text,
    compare_models,
    get_inference_device
)

__all__ = [
    "AVAILABLE_MODELS",
    "MODEL_META",
    "COLOR_MAP",
    "RISK_INK",
    "label2id",
    "id2label",
    "clean_text_pipeline",
    "segment_sentences",
    "clean_boilerplate_header",
    "check_pro_user_override",
    "has_high_risk_keyword",
    "determine_risk_level",
    "load_model",
    "classify_text",
    "compare_models",
    "get_inference_device"
]