Initalize GecBERTModel using pre-trained model weights

init_model.py

from gector.gec_model import GecBERTModel
VOCAB_PATH = "test_fixtures/roberta_model/vocabulary"
MODEL_PATH = "test_fixtures/roberta_model/weights.th"
# Initialize model
model = GecBERTModel(
    vocab_path=VOCAB_PATH,
    model_paths=[MODEL_PATH],
    max_len=50,
    min_len=3,
    iterations=5,
    min_error_probability=0.0,
    lowercase_tokens=0,
    model_name="roberta",
    special_tokens_fix=1,
    log=False,
    confidence=0,
    del_confidence=0,
    is_ensemble=0,
    weights=None,
)