From 62d98c02732dbdd99ea46e0aba6301a7f3be59a5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=80=9D=E5=AE=8F?= Date: Mon, 13 Jun 2022 17:30:48 +0800 Subject: [PATCH] [to #42322933] init --- modelscope/pipelines/nlp/nli_pipeline.py | 4 ++-- test.py | 5 ++++- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/modelscope/pipelines/nlp/nli_pipeline.py b/modelscope/pipelines/nlp/nli_pipeline.py index fe658c77..135f826a 100644 --- a/modelscope/pipelines/nlp/nli_pipeline.py +++ b/modelscope/pipelines/nlp/nli_pipeline.py @@ -31,8 +31,8 @@ class NLIPipeline(Pipeline): """ assert isinstance(model, str) or isinstance(model, SbertForNLI), \ 'model must be a single str or SbertForNLI' - sc_model = model if isinstance(model, - SbertForNLI) else SbertForNLI(model) + sc_model = model if isinstance( + model, SbertForNLI) else Model.from_pretrained(model) if preprocessor is None: preprocessor = NLIPreprocessor( sc_model.model_dir, diff --git a/test.py b/test.py index d0cd093b..b10a7d0b 100644 --- a/test.py +++ b/test.py @@ -9,4 +9,7 @@ tokenizer = NLIPreprocessor(model.model_dir) semantic_cls = pipeline('nli', model=model, preprocessor=tokenizer) print(type(semantic_cls)) -print(semantic_cls(input=('相反，这表明克林顿的敌人是疯子。', '四川商务职业学院商务管理在哪个校区？'))) +print( + semantic_cls( + input=('我想还有一件事也伤害到了老师的招聘，那就是他们在课堂上失去了很多的权威', + '教师在课堂上失去权威，导致想要进入这一职业的人减少了。')))