Update model download method

game.py

@@ -67,6 +67,13 @@ class Model_class:
 
     def __init__(self, lang=0, model_type="SentenceTransformer"):
 
+        if model_type == "SentenceTransformer":
+            repo_url = "git@hf.co:Jsevisal/strans_models"
+            dest_path = "config/strans_models/"
+        else:
+            repo_url = "git@hf.co: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/")
@@ -76,25 +83,25 @@ class Model_class:
         if lang == 1:
             if model_type == "word2vec":
                 self.model = KeyedVectors.load(
-                    model_path,
+                    os.path.join(model_path, "eng_w2v_model"),
                     mmap="r",
                 )
             elif model_type == "SentenceTransformer":
                 self.model = KeyedVectors.load(
-                    model_path,
+                    os.path.join(model_path, "eng_strans_model"),
                     mmap="r",
                 )
 
         else:
             if model_type == "word2vec":
                 self.model = KeyedVectors.load(
-                    model_path,
+                    os.path.join(model_path, "esp_w2v_model"),
                     mmap="r",
                 )
 
             elif model_type == "SentenceTransformer":
                 self.model = KeyedVectors.load(
-                    model_path,
+                    os.path.join(model_path, "esp_strans_model"),
                     mmap="r",
                 )