Import and use SinaTools

app.py

@@ -1,7 +1,24 @@
 from fastapi import FastAPI
+from sinatools.DataDownload.downloader import download_folder_from_hf
+from sinatools.wsd.disambiguator import disambiguate
+import os
 
 app = FastAPI()
 
+SINATOOLS_DIR = os.path.expanduser("~/.sinatools")
+os.makedirs(SINATOOLS_DIR, exist_ok=True)
+
+# Destination path
+dest_path = os.path.join(SINATOOLS_DIR)
+
+# 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(url='https://sina.birzeit.edu/four_grams.pickle', dest_path=dest_path)
+
+
 @app.get("/predict")
 def greet_json():
-    return {"Hello": "World!"}
+    salma_output = disambiguate('تمشيت بين الجداول والأنهار')
+    return {"output": salma_output}

requirements.txt

@@ -1,2 +1,3 @@
 fastapi
-uvicorn[standard]
+uvicorn[standard]
+SinaTools