Wendy commited on
Commit
cd3a7b0
·
verified ·
1 Parent(s): 183355e

Upload cogagent_infer_batch.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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.loads(f)
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)