Upload cogagent_infer_batch.py with huggingface_hub

cogagent_infer_batch.py

@@ -11,9 +11,9 @@ from torch.utils.data import Dataset, DataLoader
 
 class AITM_Dataset(Dataset):
     def __init__(self, json_path):
-        self.data = []
+        #self.data = []
         with open(json_path, 'r') as f:
-            self.data.append(json.loads(f))
+            self.data = json.loads(f)
 
     def __len__(self):
         return len(self.data)