baseplate
/

instructor-large-1

Model card Files Files and versions Community

andrewluo commited on Mar 27, 2023

Commit

5761c76

•

1 Parent(s): eb38ce4

Update handler.py

Files changed (1) hide show

handler.py +3 -3

handler.py CHANGED Viewed

@@ -17,6 +17,6 @@ class EndpointHandler():
         # get inputs
         instruction = data.pop("instruction",data)
         text = data.pop("text", data)
-        embeddings = self.model.encode([[instruction,text]])
-        return embeddings[0].tolist()

         # get inputs
         instruction = data.pop("instruction",data)
         text = data.pop("text", data)
+        inputs = [(s, instruction) for s in text]
+        embeddings = self.model.encode(inputs)
+        return embeddings.tolist()