Load one_gram pickle file

app.py

@@ -1,18 +1,34 @@
 from fastapi import FastAPI
-#from sinatools.DataDownload.downloader import download_folder_from_hf
+from huggingface_hub import snapshot_download
+from huggingface_hub import hf_hub_download
 import os
 
 print("Version 1")
 app = FastAPI()
 
-SINATOOLS_DIR = os.path.expanduser("~/.sinatools")
-os.makedirs(SINATOOLS_DIR, exist_ok=True)
 
-# Destination path
-dest_path = os.path.join(SINATOOLS_DIR)
+def download_file_from_hf(repo_id, filename):
+    """
+    Downloads a single file from a Hugging Face repo into ~/.sinatools
 
-from huggingface_hub import snapshot_download
-import os
+    Args:
+        repo_id (str): Hugging Face repo id, e.g. "SinaLab/ArabGlossBERT"
+        filename (str): Path of the file inside the repo, e.g. "config.json"
+
+    Returns:
+        str: Absolute path to the downloaded file
+    """
+    target_dir = os.path.expanduser("~/.sinatools")
+    os.makedirs(target_dir, exist_ok=True)
+
+    file_path = hf_hub_download(
+        repo_id=repo_id,
+        filename=filename,
+        local_dir=target_dir,
+        local_dir_use_symlinks=False
+    )
+
+    return file_path
 
 def download_folder_from_hf(repo_id, folder_name):
     """
@@ -34,10 +50,10 @@ 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_file_from_hf("SinaLab/ArabGlossBERT","one_gram.pickle")
 
 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")