bert embeddings python example