baseplate
/

instructor-large-1

Model card Files Files and versions

andrewluo commited on Mar 27, 2023

Commit

d286efc

·

1 Parent(s): 5761c76

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -17,6 +17,6 @@ class EndpointHandler():
         # 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()

         # 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()