add custom handler and modify pipeline

Files changed (4) hide show

__pycache__/bert_for_sequence_classification.cpython-37.pyc CHANGED Viewed

Binary files a/__pycache__/bert_for_sequence_classification.cpython-37.pyc and b/__pycache__/bert_for_sequence_classification.cpython-37.pyc differ

__pycache__/bibert_multitask_classification.cpython-37.pyc CHANGED Viewed

Binary files a/__pycache__/bibert_multitask_classification.cpython-37.pyc and b/__pycache__/bibert_multitask_classification.cpython-37.pyc differ

__pycache__/handler.cpython-37.pyc CHANGED Viewed

Binary files a/__pycache__/handler.cpython-37.pyc and b/__pycache__/handler.cpython-37.pyc differ

handler.py CHANGED Viewed

@@ -46,9 +46,13 @@ class EndpointHandler():
     Return:
         A :obj:`list` | `dict`: will be serialized and returned
     """
-    logger.info("INFO")
     inputs = data.pop("inputs", data)
     lang = data.pop("lang", None)
     prediction_p = self.classifier_p(inputs)
     label = prediction_p[0]['label']

     Return:
         A :obj:`list` | `dict`: will be serialized and returned
     """
     inputs = data.pop("inputs", data)
     lang = data.pop("lang", None)
+    logger.info(inputs)
+    logger.info(lang)
+    if isinstance(inputs, str):
+      inputs = [inputs]
     prediction_p = self.classifier_p(inputs)
     label = prediction_p[0]['label']