Update model download method

game.py

@@ -55,6 +55,8 @@ from tracking import (
 )
 from sentence_transformers import SentenceTransformer, util
 import warnings
+from huggingface_hub import snapshot_download
+
 
 warnings.filterwarnings(action="ignore", category=UserWarning, module="gensim")
 
@@ -65,45 +67,34 @@ class Model_class:
 
     def __init__(self, lang=0, model_type="SentenceTransformer"):
 
-        if model_type == "SentenceTransformer":
-            repo_url = "[email protected]:Jsevisal/strans_models"
-            dest_path = "config/strans_models/"
-        else:
-            repo_url = "[email protected]:Jsevisal/w2v_models"
-            dest_path = "config/w2v_models/"
-
         # Check if the model exists, clone it if it doesn't
         if not os.path.exists(
             os.path.join(self.base_path, "config/strans_models/")
         ) or not os.path.exists(os.path.join(self.base_path, "config/w2v_models/")):
-            os.system(f"git clone {repo_url} {dest_path}")
+            model_path = snapshot_download(repo_id="Jsevisal/strans_models")
 
         if lang == 1:
             if model_type == "word2vec":
                 self.model = KeyedVectors.load(
-                    os.path.join(self.base_path, "config/w2v_models/eng_w2v_model"),
+                    model_path,
                     mmap="r",
                 )
             elif model_type == "SentenceTransformer":
                 self.model = KeyedVectors.load(
-                    os.path.join(
-                        self.base_path, "config/strans_models/eng_strans_model"
-                    ),
+                    model_path,
                     mmap="r",
                 )
 
         else:
             if model_type == "word2vec":
                 self.model = KeyedVectors.load(
-                    os.path.join(self.base_path, "config/w2v_models/esp_w2v_model"),
+                    model_path,
                     mmap="r",
                 )
 
             elif model_type == "SentenceTransformer":
                 self.model = KeyedVectors.load(
-                    os.path.join(
-                        self.base_path, "config/strans_models/esp_strans_model"
-                    ),
+                    model_path,
                     mmap="r",
                 )