migration

linto-ai · Mar 31, 2023 · d3db25c · d3db25c
1 parent 3e51d9d
commit d3db25c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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'