Update handler.py

handler.py

@@ -22,5 +22,5 @@ class EndpointHandler():
             outputs = self.model(**inputs)
         
         last_hidden_states = outputs.last_hidden_state
-        return last_hidden_states[2].tolist()
+        return last_hidden_states.tolist()