t12e
/

instructor-base

t12e commited on May 30, 2023

Commit

41c2439

1 Parent(s): 7bfffff

updated handler.py

Files changed (2) hide show

__pycache__/handler.cpython-310.pyc ADDED Viewed

Binary file (926 Bytes). View file

handler.py CHANGED Viewed

@@ -10,4 +10,10 @@ class EndpointHandler():
         document = data.get("document", "")
         embedding = self.model.encode([[instruction, document]]).flatten()
-        return [{"embedding": embedding}]

         document = data.get("document", "")
         embedding = self.model.encode([[instruction, document]]).flatten()
+        return [
+            {
+                "embedding": embedding,
+                "instruction": instruction,
+                "document": document
+            }
+        ]