File size: 1,352 Bytes
2c20923
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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