updated handler.py

handler.py

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