Upload cogagent_infer.py with huggingface_hub
Browse files- cogagent_infer.py +1 -1
cogagent_infer.py
CHANGED
|
@@ -124,7 +124,7 @@ def main():
|
|
| 124 |
with open(args.input_json, "r") as f:
|
| 125 |
data = json.load(f)
|
| 126 |
res = []
|
| 127 |
-
for i in tqdm(range(
|
| 128 |
x = data[i]
|
| 129 |
img_path = x['image']
|
| 130 |
image = Image.open(img_path).convert("RGB")
|
|
|
|
| 124 |
with open(args.input_json, "r") as f:
|
| 125 |
data = json.load(f)
|
| 126 |
res = []
|
| 127 |
+
for i in tqdm(range(len(data))):
|
| 128 |
x = data[i]
|
| 129 |
img_path = x['image']
|
| 130 |
image = Image.open(img_path).convert("RGB")
|