Update Preprocess.py

Preprocess.py

@@ -55,7 +55,7 @@ class Preprocess :
         # pooled embedding of shape <1, hidden_size>
         mean_pooled_embedding = last_hidden_states.mean(axis=1)
 
-        result = mean_pooled_embedding.flatten().cpu().detach().numpy()
+        result = mean_pooled_embedding.mean(axis=0).flatten().cpu().detach().numpy()
         # print(result.shape)
         # print(result)
         # Convert the list to JSON