update handler

handler.py

@@ -16,13 +16,14 @@ class EndpointHandler:
         # get inputs
         inputs: dict = data.pop("inputs", data)
 
-        print(inputs)
+        # make sure texts is a list
+        if not isinstance(inputs, list):
+            inputs = [inputs]
 
-        # add instruction
-        query  = [['Retrieve documents that can help answer the question:',
-                   inputs]]
+        instruction = 'Retrieve documents that can help answer the question:'
 
-        # encode
-        embedding = self.model.encode(query)
+        instructions = [[instruction, text] for text in inputs]
+        
+        embeddings = self.model.encode(instructions)
 
-        return embedding
+        return embeddings.tolist()