Wendy commited on
Commit
2f0ef24
·
verified ·
1 Parent(s): 092cf2c

Upload cogagent_infer.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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(1)):
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")