From e184ad13cf9494f3235d713f945d9f944576e62d Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Wed, 15 Jan 2020 15:43:44 +0000 Subject: [PATCH] Close #2392 --- src/transformers/pipelines.py | 7 ++++--- tests/test_pipelines.py | 8 ++++---- 2 files changed, 8 insertions(+), 7 deletions(-) diff --git a/src/transformers/pipelines.py b/src/transformers/pipelines.py index 71f851b2e1..3049ff85e9 100755 --- a/src/transformers/pipelines.py +++ b/src/transformers/pipelines.py @@ -883,10 +883,10 @@ SUPPORTED_TASKS = { "pt": AutoModelForTokenClassification if is_torch_available() else None, "default": { "model": { - "pt": "https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-pytorch_model.bin", - "tf": "https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-tf_model.h5", + "pt": "dbmdz/bert-large-cased-finetuned-conll03-english", + "tf": "dbmdz/bert-large-cased-finetuned-conll03-english", }, - "config": "https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-config.json", + "config": "dbmdz/bert-large-cased-finetuned-conll03-english", "tokenizer": "bert-large-cased", }, }, @@ -925,6 +925,7 @@ def pipeline( pipeline('sentiment-analysis') pipeline('question-answering', model='distilbert-base-uncased-distilled-squad', tokenizer='bert-base-cased') pipeline('ner', model=AutoModel.from_pretrained(...), tokenizer=AutoTokenizer.from_pretrained(...) + pipeline('ner', model='dbmdz/bert-large-cased-finetuned-conll03-english', tokenizer='bert-base-cased') pipeline('ner', model='https://...pytorch-model.bin', config='https://...config.json', tokenizer='bert-base-cased') """ # Retrieve the task diff --git a/tests/test_pipelines.py b/tests/test_pipelines.py index 43cc1f2474..936fbd8813 100644 --- a/tests/test_pipelines.py +++ b/tests/test_pipelines.py @@ -21,16 +21,16 @@ TF_QA_FINETUNED_MODELS = { TF_NER_FINETUNED_MODELS = { ( "bert-base-cased", - "https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-tf_model.h5", - "https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-config.json", + "dbmdz/bert-large-cased-finetuned-conll03-english", + "dbmdz/bert-large-cased-finetuned-conll03-english", ) } NER_FINETUNED_MODELS = { ( "bert-base-cased", - "https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-pytorch_model.bin", - "https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-config.json", + "dbmdz/bert-large-cased-finetuned-conll03-english", + "dbmdz/bert-large-cased-finetuned-conll03-english", ) }