fixed montenegrin

app.py

@@ -33,7 +33,7 @@ MODEL_MAP = {
     "lb": "alirezamsh/small100", # luxembourgish          # small100
     "lt": "Helsinki-NLP/opus-mt-tc-big-en-lt", # lithuanian
     "lv": "facebook/mbart-large-50-many-to-many-mmt", # latvian
-    "me": "Helsinki-NLP/opus-mt-tc-base-en-sh", # montegrin
+    "cnr": "Helsinki-NLP/opus-mt-tc-base-en-sh", # montegrin
     "mk": "Helsinki-NLP/opus-mt-en-mk", # macedonian
     # "nb": "facebook/mbart-large-50-many-to-many-mmt", # norwegian
     "nl": "facebook/mbart-large-50-many-to-many-mmt", # dutch
@@ -99,6 +99,8 @@ async def translate(request: Request):
 
             if model_id == "Confused404/eng-gmq-finetuned_v2-no":
                 chunk = f">>nob<< {chunk}"
+            if model_id == "Helsinki-NLP/opus-mt-tc-base-en-sh":
+                chunk = f">>cnr<< {chunk}"
             
             inputs  = tokenizer(chunk, return_tensors="pt", padding=True, truncation=True)
             inputs  = {k: v.to(model.device) for k, v in inputs.items()}