WUBIAO commited on
Commit
4ef8ce0
·
verified ·
1 Parent(s): f9d46da

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.append(json.loads(line))
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.append(json.loads(f))
17
 
18
  def __len__(self):
19
  return len(self.data)