Add snapshot_download

app.py

@@ -1,7 +1,7 @@
 from fastapi import FastAPI
 import torch
 import pickle
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, snapshot_download
 from Nested.nn.BertSeqTagger import BertSeqTagger
 from transformers import AutoTokenizer, AutoModel
 import inspect
@@ -23,10 +23,12 @@ pretrained_path = "aubmindlab/bert-base-arabertv2"  # must match training
 tokenizer = AutoTokenizer.from_pretrained(pretrained_path)
 encoder = AutoModel.from_pretrained(pretrained_path).eval()
 
-checkpoint_path = hf_hub_download(
-    repo_id="SinaLab/Nested",
-    filename="checkpoints/checkpoint_2.pt"
-)
+# checkpoint_path = hf_hub_download(
+#     repo_id="SinaLab/Nested",
+#     filename="checkpoints/checkpoint_2.pt"
+# )
+
+checkpoint_path = snapshot_download(repo_id="SinaLab/Nested", allow_patterns="checkpoints/")
 print("checkpoint_path : ", checkpoint_path)
 
 args_path = hf_hub_download(