Update backPrompt.py

backPrompt.py

@@ -26,16 +26,8 @@ def load_image(image_file):
     return pixel_values
 
 
-def main(image_path):
+def main(image_path,model,tokenizer):
     path = "OpenGVLab/InternVL2_5-4B"
-    model = AutoModel.from_pretrained(
-        path,
-        torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32,
-        # load_in_8bit=True,
-        low_cpu_mem_usage=True,
-        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).to(torch.float32).to("cpu")
     generation_config = dict(max_new_tokens=1024, do_sample=True)