Update frontPrompt.py

frontPrompt.py

@@ -20,7 +20,7 @@ def build_transform(input_size):
     return transform
 
 
-def load_image(image_file, input_size=800):
+def load_image(image_file):
     transform = build_transform(input_size=input_size)
     pixel_values = transform(image_file).unsqueeze(0)  # Add batch dimension
     return pixel_values
@@ -36,7 +36,7 @@ def main(image_path):
         use_flash_attn=True,
         trust_remote_code=True).eval()
     tokenizer = AutoTokenizer.from_pretrained(path, trust_remote_code=True, use_fast=False)
-    pixel_values = load_image(image_path, max_num=12).to(torch.bfloat16).cuda()
+    pixel_values = load_image(image_path).to(torch.bfloat16).cuda()
     generation_config = dict(max_new_tokens=1024, do_sample=True)
     
     question = """<image>\n**Instruction:**