Add args_data

app.py

@@ -8,6 +8,7 @@ import inspect
 from collections import namedtuple
 from Nested.utils.helpers import load_checkpoint
 from Nested.utils.data import get_dataloaders, text2segments
+import json
 
 app = FastAPI()
 print("Version 2...")
@@ -38,6 +39,11 @@ args_path = hf_hub_download(
     filename="args.json"
 )
 
+with open(args_path, 'r') as f:
+    args_data = json.load(f)
+    
+print("data : ", args_data)
+
 # Load model
 with open("Nested/utils/tag_vocab.pkl", "rb") as f:
     label_vocab = pickle.load(f)
@@ -60,7 +66,7 @@ vocab = vocabs(tokens=token_vocab, tags=tag_vocab)
 dataloader = get_dataloaders(
     (dataset,),
     vocab,
-    args_path,
+    args_data,
     batch_size=32,
     shuffle=(False,),
 )[0]