Apppublic

Spico/Mirror

sourceHugging Faceapache-2.0updated 6d agoView on Hugging Face

4likes

inference.py24 linesDownload Raw Back to src

1import os2 3from rex.utils.logging import logger4 5from src.task import MrcTaggingTask6 7if __name__ == "__main__":8    os.environ["CUDA_VISIBLE_DEVICES"] = ""9 10    task = MrcTaggingTask.from_taskdir(11        "outputs/bert_mrc_ner",12        load_best_model=True,13        update_config={14            "skip_train": True,15            "debug_mode": False,16        },17    )18 19    cases = ["123123", "123123"]20    logger.info(f"Cases: {cases}")21 22    ents = task.predict(cases)23    logger.info(f"Results: {ents}")24