from item_embedding.all_lang_emb import item2emb_all from item_embedding.chinese_emb import item2emb_cn from item_embedding.eng_emb import item2emb_eng def main_emb(sentences, lang="all"): embedings = [] if lang == "all": embedings = item2emb_all(sentences) elif lang == "cn": embedings = item2emb_cn(sentences) elif lang == "eng": embedings = item2emb_eng(sentences) return embedings