Spaces:
Running
on
T4
Running
on
T4
File size: 1,352 Bytes
5b5a888 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
import os
LID_LANGUAGES = {
"eng_Latn": "english",
'zho_Hans': "chinese",
'zho_Hant': "chinese",
'deu_Latn': "german",
'spa_Latn': "spanish",
'rus_Cyrl': "russian",
'kor_Hang': "korean",
'fra_Latn': "french",
'jpn_Jpan': "japanese",
'por_Latn': "portuguese",
'tur_Latn': "turkish",
'pol_Latn': "polish",
'nld_Latn': "dutch",
'arb_Arab': "arabic",
'ita_Latn': "italian",
'ind_Latn': "indonesian",
'hin_Deva': "hindi",
'vie_Latn': "vietnamese",
'heb_Hebr': "hebrew",
'ukr_Cyrl': "ukrainian",
'ell_Grek': "greek",
'ces_Latn': "czech",
'ron_Latn': "romanian",
'pes_Arab': "persian",
}
NEETS_AI_LANGID_MAP={
'english': 'eng-1',
'chinese': 'zho-1',
'german': 'deu-1',
'spanish': 'spa-1',
'russian': 'rus-1',
'korean': 'kor-1',
'french': 'fra-1',
'portuguese': 'por-2',
'turkish': 'tur-1',
'polish': 'pol-2',
'dutch': 'nld-3',
'arabic': 'ara-1',
'italian': 'ita-1',
'indonesian': 'ind-1',
'hindi': 'hin-1',
'vietnamese': 'vie-1',
'hebrew': 'heb-1',
'ukrainian': 'ukr-1',
'greek': 'ell-1',
'czech': 'cze-1',
'romanian': 'ron-1',
'persian': 'fas-1'
}
AYA_MODEL_NAME = os.getenv("AYA_MODEL_NAME")
BATCH_SIZE = 8
USE_REPLICATE = True
USE_ELVENLABS = False |