HCKLab
/

BiBert-MultiTask-1

Text Classification

Generated from Trainer

text-embeddings-inference

Model card Files Files and versions

Metrics Training metrics Community

Kalaoke commited on Oct 24, 2022

Commit

cc6c3a1

·

1 Parent(s): 595bda0

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -251,7 +251,7 @@ class EndpointHandler():
         A :obj:`list` | `dict`: will be serialized and returned
     """
     inputs = data.pop("text", data)
-    lang = data.pop("lang", data)
     if isinstance(inputs, str):
       inputs = [inputs]

         A :obj:`list` | `dict`: will be serialized and returned
     """
     inputs = data.pop("text", data)
+    lang = data.pop("lang", None)
     if isinstance(inputs, str):
       inputs = [inputs]