Upload cogagent_infer.py with huggingface_hub

cogagent_infer.py

@@ -8,6 +8,7 @@ from typing import List
 import json
 from tqdm import tqdm
 
+class 
 def draw_boxes_on_image(image: Image.Image, boxes: List[List[float]], save_path: str):
     """
     Draws red bounding boxes on the given image and saves it.
@@ -123,7 +124,7 @@ def main():
     with open(args.input_json, "r") as f:
         data = json.load(f)
     res = []
-    for i in tqdm(range(len(data))):
+    for i in tqdm(range(5)):
         x = data[i]
         img_path = x['image']
         image = Image.open(img_path).convert("RGB")