blancsw
/

lang-detection

Text Classification

text-embeddings-inference

Model card Files Files and versions

blancsw commited on Feb 16, 2024

Commit

2c17eab

·

verified ·

1 Parent(s): df07b9f

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -33,4 +33,4 @@ class EndpointHandler:
         else:
             prediction = self.pipline(inputs)
         # postprocess the prediction
-        return prediction

         else:
             prediction = self.pipline(inputs)
         # postprocess the prediction
+        return [{"label": p["label"]} for p in prediction]