en2cn/item_embedding/main_emb.py

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