making the cap model smaller

interfaces/cap.py

@@ -14,19 +14,8 @@ from label_dicts import CAP_NUM_DICT, CAP_LABEL_NAMES
 HF_TOKEN = os.environ["hf_read"]
 
 languages = [
-    "Danish",
-    "Dutch",
     "English",
-    "French",
-    "German",
-    "Hungarian",
-    "Italian",
-    "Polish",
-    "Portuguese",
-    "Spanish",
-    "Czech",
-    "Slovak",
-    "Norwegian"
+    "Multilingual"
 ]
 
 domains = {
@@ -76,21 +65,13 @@ def build_huggingface_path(language: str, domain: str):
     except (AttributeError, FileNotFoundError):
         value = None
 
-    if value and value in path_map:
-        model_path = path_map[value]
-        if check_huggingface_path(model_path):
-            # if the model is available on Huggingface, return the path
-            return model_path
-        else:
-            # if the model is not available on Huggingface, look for other models
-            filtered_path_map = {k: v for k, v in path_map.items() if k != value}
-            for k, v in filtered_path_map.items():
-                if check_huggingface_path(v):
-                    return v
-    elif check_huggingface_path(lang_domain_path):
-        return lang_domain_path
-    elif check_huggingface_path(lang_path):
-        return lang_path
+    if language == 'english':
+        model_path = lang_path
+    else:
+        model_path = "poltextlab/xlm-roberta-large-pooled-cap"
+
+    if check_huggingface_path(model_path):
+        return model_path
     else:
         return "poltextlab/xlm-roberta-large-pooled-cap"