Wendy
commited on
Upload cogagent_infer_batch.py with huggingface_hub
Browse files- cogagent_infer_batch.py +1 -1
cogagent_infer_batch.py
CHANGED
@@ -13,7 +13,7 @@ class AITM_Dataset(Dataset):
|
|
13 |
def __init__(self, json_path):
|
14 |
#self.data = []
|
15 |
with open(json_path, 'r') as f:
|
16 |
-
self.data = json.
|
17 |
|
18 |
def __len__(self):
|
19 |
return len(self.data)
|
|
|
13 |
def __init__(self, json_path):
|
14 |
#self.data = []
|
15 |
with open(json_path, 'r') as f:
|
16 |
+
self.data = json.load(f)
|
17 |
|
18 |
def __len__(self):
|
19 |
return len(self.data)
|