Upload cogagent_infer.py with huggingface_hub

cogagent_infer.py

@@ -6,6 +6,7 @@ from PIL import Image, ImageDraw
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 from typing import List
 import json
+form tqdm import tqdm
 
 def draw_boxes_on_image(image: Image.Image, boxes: List[List[float]], save_path: str):
     """
@@ -122,7 +123,8 @@ def main():
     with open(args.input_json, "r") as f:
         data = json.load(f)
     res = []
-    for x in data:
+    for i in tqdm(range(len(data))):
+        x = data[i]
         img_path = x['image']
         image = Image.open(img_path).convert("RGB")
         task = x['conversations'][0]['value']
@@ -189,7 +191,7 @@ def main():
             draw_boxes_on_image(image, boxes, output_path)
             #print(f"Annotated image saved at: {output_path}")
         ans = {
-            'query': query,
+            'query': f"Round {round_num} query:\n{query}",
             'response': response,
             'output_path': output_path
         }