build huggingface path (tuple)

interfaces/cap_minor.py

@@ -69,10 +69,21 @@ def check_huggingface_path(checkpoint_path: str):
         return False
 
 
-def build_huggingface_path(language: str, domain: str):
-    if domain in ["social"]:
-        return "poltextlab/xlm-roberta-large-twitter-cap-minor"
-    return "poltextlab/xlm-roberta-large-pooled-cap-minor-v3"
+def build_huggingface_path(language: str, domain: str, hierarchical: bool):
+    if hierarchical:
+        major = "poltextlab/xlm-roberta-large-pooled-cap-v3"
+        minor = (
+            "poltextlab/xlm-roberta-large-twitter-cap-minor"
+            if domain == "social"
+            else "poltextlab/xlm-roberta-large-pooled-cap-minor-v3"
+        )
+        return major, minor
+
+    return (
+        "poltextlab/xlm-roberta-large-twitter-cap-minor"
+        if domain == "social"
+        else "poltextlab/xlm-roberta-large-pooled-cap-minor-v3"
+    )
 
 
 def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):