Update handler.py

bug: fix an issue where the response returns dimensions of (1, 3, 1024) for a single string. This code was missing the edits made to the usage code on https://huggingface.co/intfloat/e5-large-v2

handler.py

@@ -24,6 +24,6 @@ class EndpointHandler():
         outputs = self.model(**batch_dict)
 
         embeddings = average_pool(outputs.last_hidden_state, batch_dict['attention_mask'])
-        embeddings = F.normalize(embeddings, p=2, dim=1).tolist()
-
-        return embeddings
+        embeddings = F.normalize(embeddings, p=2, dim=1)
+        scores = (embeddings[:2] @ embeddings[2:].T) * 100
+        return embeddings.tolist()