Update

app.py

@@ -1,6 +1,5 @@
 from fastapi import FastAPI
 #from sinatools.DataDownload.downloader import download_folder_from_hf
-from sinatools.wsd.disambiguator import disambiguate
 import os
 
 print("Version 1")
@@ -15,29 +14,31 @@ dest_path = os.path.join(SINATOOLS_DIR)
 from huggingface_hub import snapshot_download
 import os
 
-def download_folder_from_hf():
-    local_dir = os.path.expanduser("~/.sinatools")
+def download_folder_from_hf(repo_id, folder_name):
+    """
+    Downloads a folder from a Hugging Face model repo into ~/.sinatools
+    """
+    target_dir = os.path.expanduser("~/.sinatools")
 
-    snapshot_download(
-        repo_id="SinaLab/ArabGlossBERT",
-        allow_patterns="bert-base-arabertv02/**",
-        local_dir=local_dir,
+    local_path = snapshot_download(
+        repo_id=repo_id,
+        allow_patterns=f"{folder_name}/**",
+        local_dir=target_dir,
         local_dir_use_symlinks=False
     )
 
-    return os.path.join(local_dir, "bert-base-arabertv02")
+    return os.path.join(local_path, folder_name)
 
 
 print("Start loading")
 # Download explicitly to the expected path
-# download_folder_from_hf("SinaLab/ArabGlossBERT", "bert-base-arabertv02_22_May_2021_00h_allglosses_unused01")
-# download_folder_from_hf("SinaLab/ArabGlossBERT", "bert-base-arabertv02")   
-download_folder_from_hf()
+download_folder_from_hf("SinaLab/ArabGlossBERT", "bert-base-arabertv02_22_May_2021_00h_allglosses_unused01")
+download_folder_from_hf("SinaLab/ArabGlossBERT", "bert-base-arabertv02")   
 
 print("Finish loading")
 
 # download_file(url='https://sina.birzeit.edu/four_grams.pickle', dest_path=dest_path)
-
+from sinatools.wsd.disambiguator import disambiguate
 
 @app.get("/predict")
 def greet_json():