Update app.py

app.py

@@ -24,8 +24,27 @@ if not os.path.exists(MODEL_DIR) or not os.listdir(MODEL_DIR):
         local_dir_use_symlinks=False
     )
 
-snapshot_download(repo_id="SinaLab/Nested")
+
+pretrained_path = "aubmindlab/bert-base-arabertv2"  # must match training
+tokenizer = AutoTokenizer.from_pretrained(pretrained_path)
+encoder = AutoModel.from_pretrained(pretrained_path).eval()
+
+
+checkpoint_path = snapshot_download(repo_id="SinaLab/Nested", allow_patterns="checkpoints/")
+
+args_path = hf_hub_download(
+    repo_id="SinaLab/Nested",
+    filename="args.json"
+)
+
+with open(args_path, 'r') as f:
+    args_data = json.load(f)
     
+# Load model
+with open("Nested/utils/tag_vocab.pkl", "rb") as f:
+    label_vocab = pickle.load(f)
+
+
 from sinatools.relations.relation_extractor import relation_extraction
 from sinatools.relations.event_relation_extractor import event_argument_relation_extraction