languages listed
Browse files
app.py
CHANGED
|
@@ -16,35 +16,35 @@ app = FastAPI()
|
|
| 16 |
|
| 17 |
# Map target languages to Hugging Face model IDs
|
| 18 |
MODEL_MAP = {
|
| 19 |
-
"bg": "Helsinki-NLP/opus-mt-tc-big-en-bg",
|
| 20 |
-
"cs": "Helsinki-NLP/opus-mt-en-cs",
|
| 21 |
-
"da": "Helsinki-NLP/opus-mt-en-da",
|
| 22 |
-
"de": "Helsinki-NLP/opus-mt-en-de",
|
| 23 |
-
"el": "Helsinki-NLP/opus-mt-tc-big-en-el",
|
| 24 |
-
"es": "Helsinki-NLP/opus-mt-tc-big-en-es",
|
| 25 |
-
"et": "Helsinki-NLP/opus-mt-tc-big-en-et",
|
| 26 |
-
"fi": "Helsinki-NLP/opus-mt-tc-big-en-fi",
|
| 27 |
-
"fr": "Helsinki-NLP/opus-mt-en-fr",
|
| 28 |
-
"hr": "facebook/mbart-large-50-many-to-many-mmt",
|
| 29 |
-
"hu": "Helsinki-NLP/opus-mt-tc-big-en-hu",
|
| 30 |
-
"is": "mkorada/opus-mt-en-is-finetuned-v4", # Manas's fine-tuned model
|
| 31 |
-
"it": "Helsinki-NLP/opus-mt-tc-big-en-it",
|
| 32 |
-
"lb": "alirezamsh/small100",
|
| 33 |
-
"lt": "Helsinki-NLP/opus-mt-tc-big-en-lt",
|
| 34 |
-
"lv": "facebook/mbart-large-50-many-to-many-mmt",
|
| 35 |
-
"me": "Helsinki-NLP/opus-mt-tc-base-en-sh",
|
| 36 |
-
"mk": "Helsinki-NLP/opus-mt-en-mk",
|
| 37 |
-
"nb": "facebook/mbart-large-50-many-to-many-mmt",
|
| 38 |
-
"nl": "facebook/mbart-large-50-many-to-many-mmt",
|
| 39 |
-
"no": "Confused404/eng-gmq-finetuned_v2-no", # Alex's fine-tuned model
|
| 40 |
-
"pl": "Helsinki-NLP/opus-mt-en-sla",
|
| 41 |
-
"pt": "facebook/mbart-large-50-many-to-many-mmt",
|
| 42 |
-
"ro": "facebook/mbart-large-50-many-to-many-mmt",
|
| 43 |
-
"sk": "Helsinki-NLP/opus-mt-en-sk",
|
| 44 |
-
"sl": "alirezamsh/small100",
|
| 45 |
-
"sq": "alirezamsh/small100",
|
| 46 |
-
"sv": "Helsinki-NLP/opus-mt-en-sv",
|
| 47 |
-
"tr": "Helsinki-NLP/opus-mt-tc-big-en-tr"
|
| 48 |
}
|
| 49 |
|
| 50 |
# Cache loaded models/tokenizers
|
|
|
|
| 16 |
|
| 17 |
# Map target languages to Hugging Face model IDs
|
| 18 |
MODEL_MAP = {
|
| 19 |
+
"bg": "Helsinki-NLP/opus-mt-tc-big-en-bg", # bulgarian
|
| 20 |
+
"cs": "Helsinki-NLP/opus-mt-en-cs", # czech
|
| 21 |
+
"da": "Helsinki-NLP/opus-mt-en-da", # danish
|
| 22 |
+
"de": "Helsinki-NLP/opus-mt-en-de", # german
|
| 23 |
+
"el": "Helsinki-NLP/opus-mt-tc-big-en-el", # greek
|
| 24 |
+
"es": "Helsinki-NLP/opus-mt-tc-big-en-es", # spanish
|
| 25 |
+
"et": "Helsinki-NLP/opus-mt-tc-big-en-et", # estonian
|
| 26 |
+
"fi": "Helsinki-NLP/opus-mt-tc-big-en-fi", # finnish
|
| 27 |
+
"fr": "Helsinki-NLP/opus-mt-en-fr", # french
|
| 28 |
+
"hr": "facebook/mbart-large-50-many-to-many-mmt", # croatian
|
| 29 |
+
"hu": "Helsinki-NLP/opus-mt-tc-big-en-hu", # hungarian
|
| 30 |
+
"is": "mkorada/opus-mt-en-is-finetuned-v4", # icelandic # Manas's fine-tuned model
|
| 31 |
+
"it": "Helsinki-NLP/opus-mt-tc-big-en-it", # italian
|
| 32 |
+
"lb": "alirezamsh/small100", # luxembourgish # small100
|
| 33 |
+
"lt": "Helsinki-NLP/opus-mt-tc-big-en-lt", # lithuanian
|
| 34 |
+
"lv": "facebook/mbart-large-50-many-to-many-mmt", # latvian
|
| 35 |
+
"me": "Helsinki-NLP/opus-mt-tc-base-en-sh", # montegrin
|
| 36 |
+
"mk": "Helsinki-NLP/opus-mt-en-mk", # macedonian
|
| 37 |
+
# "nb": "facebook/mbart-large-50-many-to-many-mmt", # norwegian
|
| 38 |
+
"nl": "facebook/mbart-large-50-many-to-many-mmt", # dutch
|
| 39 |
+
"no": "Confused404/eng-gmq-finetuned_v2-no", # norwegian # Alex's fine-tuned model
|
| 40 |
+
"pl": "Helsinki-NLP/opus-mt-en-sla", # polish
|
| 41 |
+
"pt": "facebook/mbart-large-50-many-to-many-mmt", # portuguese
|
| 42 |
+
"ro": "facebook/mbart-large-50-many-to-many-mmt", # romanian
|
| 43 |
+
"sk": "Helsinki-NLP/opus-mt-en-sk", # slovak
|
| 44 |
+
"sl": "alirezamsh/small100", # slovene
|
| 45 |
+
"sq": "alirezamsh/small100", # albanian
|
| 46 |
+
"sv": "Helsinki-NLP/opus-mt-en-sv", # swedish
|
| 47 |
+
"tr": "Helsinki-NLP/opus-mt-tc-big-en-tr" # turkish
|
| 48 |
}
|
| 49 |
|
| 50 |
# Cache loaded models/tokenizers
|