import torchfrom PIL import Imagefrom transformers import AutoProcessor, Qwen2_5_VLForConditionalGenerationfrom peft import PeftModelfrom qwen_vl_utils import process_vision_info # Para Qwen2.5-VL-7B use subfolder="qwen2_5_vl_7b" e base "Qwen/Qwen2.5-VL-7B-Instruct"# Para Qwen3-VL-8B use subfolder="qwen3_vl_8b" e base "Qwen/Qwen3-VL-8B-Instruct" subfolder = "qwen2_5_vl_7b"base_model_name = "Qwen/Qwen2.5-VL-7B-Instruct" processor = AutoProcessor.from_pretrained("lucasoc/sci-image-models", subfolder=subfolder)base_model = Qwen2_5_VLForConditionalGeneration.from_pretrained( base_model_name, torch_dtype=torch.float16, device_map="auto")model = PeftModel.from_pretrained(base_model, "lucasoc/sci-image-models", subfolder=subfolder)model.eval() # Processamento da figuraimage = Image.open("exemplo_grafico_cientifico.png").convert("RGB")messages = [ { "role": "user", "content": [ {"type": "image", "image": image}, {"type": "text", "text": "Extract all numerical data points from this scientific figure into a Markdown table."} ] }]text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)image_inputs, _ = process_vision_info(messages)inputs = processor(text=[text], images=image_inputs, return_tensors="pt").to("cuda") with torch.inference_mode(): outputs = model.generate(**inputs, max_new_tokens=1024) result = processor.batch_decode(outputs, skip_special_tokens=True)[0] print(result)