Revert "[DEBUGONLY] Try returning tokens"

This reverts commit b957c4dfa1a7e60027ef0185247b32dfedd1aba7.

handler.py

@@ -28,5 +28,10 @@ class EndpointHandler():
         inputs = data.get("inputs", data)
 
         # tokenize the input
-        encoded_inputs = self.tokenizer(inputs, padding=True, truncation=True)
-        return {"tokens": encoded_inputs.data}
+        encoded_inputs = self.tokenizer(inputs, padding=True, truncation=True, return_tensors='pt')
+        # run the model
+        outputs = self.model(**encoded_inputs)
+        # Perform pooling
+        sentence_embeddings = mean_pooling(outputs, encoded_inputs['attention_mask'])
+        # postprocess the prediction
+        return sentence_embeddings.tolist()