Wendy commited on
Commit
84e1125
·
verified ·
1 Parent(s): 0fa8989

Upload cogagent_infer.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. cogagent_infer.py +1 -1
cogagent_infer.py CHANGED
@@ -6,7 +6,7 @@ from PIL import Image, ImageDraw
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
7
  from typing import List
8
  import json
9
- form tqdm import tqdm
10
 
11
  def draw_boxes_on_image(image: Image.Image, boxes: List[List[float]], save_path: str):
12
  """
 
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
7
  from typing import List
8
  import json
9
+ from tqdm import tqdm
10
 
11
  def draw_boxes_on_image(image: Image.Image, boxes: List[List[float]], save_path: str):
12
  """