fix norm

handler.py

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