migration

Author: guokan-shang

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():

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'