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.append(json.loads(
|
| 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)
|