Wendy-Fly commited on
Commit
b7eccab
·
verified ·
1 Parent(s): 8b05886

Upload infer_qwen2_vl.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. infer_qwen2_vl.py +1 -0
infer_qwen2_vl.py CHANGED
@@ -27,6 +27,7 @@ processor = AutoProcessor.from_pretrained(model_path)
27
  import glob
28
  from PIL import Image
29
  import argparse
 
30
 
31
  # parser = argparse.ArgumentParser(description="Process a dataset with specific index range.")
32
  # parser.add_argument("--batch_size", type=int, default = 1,help="batch size")
 
27
  import glob
28
  from PIL import Image
29
  import argparse
30
+ import os
31
 
32
  # parser = argparse.ArgumentParser(description="Process a dataset with specific index range.")
33
  # parser.add_argument("--batch_size", type=int, default = 1,help="batch size")