Upload cogagent_infer_batch.py with huggingface_hub

cogagent_infer_batch.py

@@ -13,8 +13,7 @@ class AITM_Dataset(Dataset):
     def __init__(self, json_path):
         self.data = []
         with open(json_path, 'r') as f:
-            for line in f:
-                self.data.append(json.loads(line))
+            self.data.append(json.loads(line))
 
     def __len__(self):
         return len(self.data)
@@ -219,10 +218,12 @@ def main():
         }
         res.append(ans)
         round_num += 1
-    print(res)
+    #print(res)
+    print("Writing to json file")
     with open(args.output_json, "w") as file:
         print("Writing to json file")
         json.dump(res, file, indent=4)
+    print("Done")