Views
No views yet
1from transformers import AutoProcessor, Qwen2VLForConditionalGeneration
2from PIL import Image
3import torch
4
5model_id = "Devildarker6789/qwen2vl-chartqa"
6
7processor = AutoProcessor.from_pretrained(model_id, min_pixels=256*28*28, max_pixels=512*28*28)
8model = Qwen2VLForConditionalGeneration.from_pretrained(
9 model_id,
10 torch_dtype=torch.float16,
11 device_map="auto"
12)
13model.eval()
14
15image = Image.open("your_chart.png").convert("RGB")
16question = "What is the highest value in this chart?"
17
18messages = [{"role": "user", "content": [
19 {"type": "image", "image": image},
20 {"type": "text", "text": question}
21]}]
22
23text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
24inputs = processor(text=[text], images=[image], return_tensors="pt").to(model.device)
25
26with torch.no_grad():
27 out = model.generate(**inputs, max_new_tokens=64, do_sample=False)
28
29print(processor.decode(out[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))
Then click **Save** — and your HF link is: