diff --git a/celery_app/tasks.py b/celery_app/tasks.py index 85376b3..c69ac9e 100644 --- a/celery_app/tasks.py +++ b/celery_app/tasks.py @@ -15,6 +15,7 @@ def kpe_task(lang: str, texts: List[str], component_cfg: Dict = {}): """Process a batch of articles and return the Keyphrases predicted by the given model. Each record in the data should have a key "text". """ + logger.info('KPE task received') # Check language availability if lang in LM_MAP.keys(): diff --git a/document/swagger.yml b/document/swagger.yml index 9f0edc6..a6c97da 100644 --- a/document/swagger.yml +++ b/document/swagger.yml @@ -61,8 +61,8 @@ components: properties: text: type: string - end: - score: float + score: + type: float batch: type: object properties: @@ -76,7 +76,7 @@ components: responsemodel: type: object properties: - ner: + kpe: type: array items: $ref: '#/components/schemas/batch'