out of scope variable fixed

interfaces/emotion9.py

@@ -20,15 +20,10 @@ domains = {
     "parliamentary speech": "parlspeech",
 }
 
-USE_V2 = False
-if language in ["slovak", "polish", "czech", "hungarian"]:
-    EMOTION9_LABEL_NAMES = EMOTION9_V2_LABEL_NAMES
-    USE_V2 = True
-
 def build_huggingface_path(language: str):
     language = language.lower()
     base_model_id = f"poltextlab/xlm-roberta-large-pooled-{language}-emotions9"
-    if USE_V2:
+    if language in ["slovak", "polish", "czech", "hungarian"]:
         return base_model_id + "-v2"
     return base_model_id
 
@@ -62,6 +57,8 @@ def predict(text, model_id, tokenizer_id):
 
     probs = torch.nn.functional.softmax(logits, dim=1).cpu().numpy().flatten()
 
+    if "v2" in model_id:
+        EMOTION9_LABEL_NAMES = EMOTION9_V2_LABEL_NAMES
     NUMS_DICT = {i: key for i, key in enumerate(sorted(EMOTION9_LABEL_NAMES.keys()))}
     output_pred = {f"[{NUMS_DICT[i]}] {EMOTION9_LABEL_NAMES[NUMS_DICT[i]]}": probs[i] for i in np.argsort(probs)[::-1]}
     output_info = f'<p style="text-align: center; display: block">Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.</p>'