jpohhhh
/

biencoder_embedding

Feature Extraction

text-embeddings-inference

Model card Files Files and versions

jpohhhh commited on Apr 29, 2023

Commit

e7eebac

·

1 Parent(s): bee946d

Fix model reference

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -14,5 +14,5 @@ class EndpointHandler():
             A :obj:`list` | `dict`: will be serialized and returned
         """
         sentences = data.pop("inputs",data)
-        embeddings = model.encode(sentences)
         return embeddings.tolist()

             A :obj:`list` | `dict`: will be serialized and returned
         """
         sentences = data.pop("inputs",data)
+        embeddings = self.model.encode(sentences)
         return embeddings.tolist()