import torchfrom transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessorfrom PIL import Imageimport requests MODEL_ID = "kiel2/Kiel-2-OCR" print("Loading Kiel-2-OCR processor and model...")processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)model = Qwen2_5_VLForConditionalGeneration.from_pretrained( MODEL_ID, torch_dtype=torch.float16, device_map="auto", trust_remote_code=True)model.eval() # Prepare an image containing textimage_url = "[https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/car.jpg](https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/transformers/tasks/car.jpg)"image = Image.open(requests.get(image_url, stream=True).raw) conversation = [ { "role": "user", "content": [ {"type": "image", "image": image}, {"type": "text", "text": "Extract all readable text from this image accurately."}, ], }] # Apply chat templatetext_prompt = processor.apply_chat_template( conversation, tokenize=False, add_generation_prompt=True) # Process inputsinputs = processor( text=[text_prompt], images=[image], padding=True, return_tensors="pt").to(model.device) # Generate responsewith torch.no_grad(): output_ids = model.generate(**inputs, max_new_tokens=512) generated_ids = [ output_ids[len(input_ids):] for input_ids, output_ids in zip(inputs.input_ids, output_ids)]response = processor.batch_decode( generated_ids, skip_special_tokens=True, clean_up_tokenization_spaces=True)[0] print("Extracted Text:\n", response)