diff --git a/combo/main.py b/combo/main.py index 9736646f6468d9c2293bada9e5397c0828fc9a79..c815130f50df92f560760fd52c29e6b09eaab955 100755 --- a/combo/main.py +++ b/combo/main.py @@ -309,9 +309,9 @@ def run(_): if not dataset_reader: logger.info("No dataset reader in the configuration or archive file - using a default UD dataset reader", prefix=prefix) - dataset_reader = default_ud_dataset_reader() + dataset_reader = default_ud_dataset_reader(FLAGS.pretrained_transformer_name) - predictor = COMBO(model, dataset_reader) + predictor = COMBO(model, dataset_reader, LamboTokenizer(language=FLAGS.tokenizer_language)) if FLAGS.input_file == '-': print("Interactive mode.") @@ -322,7 +322,11 @@ def run(_): print("{:15} {:15} {:10} {:10} {:10}".format(token.text, token.lemma, token.upostag, token.head, token.deprel)) elif FLAGS.output_file: - checks.file_exists(FLAGS.input_file) + try: + checks.file_exists(FLAGS.input_file) + except ConfigurationError as e: + handle_error(e, prefix) + logger.info("Predicting examples from file", prefix=prefix) if FLAGS.conllu_format: