Test NER model

app.py

@@ -22,13 +22,10 @@ download_file(url='https://sina.birzeit.edu/two_grams.pickle', dest_path=dest_pa
 
 from sinatools.morphology.morph_analyzer import analyze
 
-# download_file('https://sina.birzeit.edu/lemmas_dic.pickle')
-# download_folder_from_hf("TymaaHammouda/testing_lemmatizer_model", 'lemmas_dic.pickle')
 
-# args_path = hf_hub_download(
-#     repo_id="SinaLab/Nested",
-#     filename="args.json"
-# )
+download_file(url='https://sina.birzeit.edu/Wj27012000.tar.gz', dest_path=dest_path)
+from sinatools.ner.entity_extractor import extract
+
 
 @app.get("/")
 def greet_json():
@@ -36,4 +33,6 @@ def greet_json():
     output =parser.arStrip('2023الجو جميلُ')
     print(output)
     lemmatizer_results = analyze('ذهب الولد الى المدرسة')
-    return {"output": lemmatizer_results}
+
+    ner_output = extract('ذهب محمد الى جامعة بيرزيت')
+    return {"output": ner_output}