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