Views
No views yet

1from transformers import TrOCRProcessor, VisionEncoderDecoderModel
2from PIL import Image
3import requests
4
5# Load model and processor
6processor = TrOCRProcessor.from_pretrained("YOUR_USERNAME/deepseek-ocr-german")
7model = VisionEncoderDecoderModel.from_pretrained("YOUR_USERNAME/deepseek-ocr-german")
8
9# Load image
10url = "path_to_your_german_text_image.jpg"
11image = Image.open(url).convert("RGB")
12
13# Process
14pixel_values = processor(image, return_tensors="pt").pixel_values
15generated_ids = model.generate(pixel_values)
16generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
17
18print(generated_text)1from transformers import TrOCRProcessor, VisionEncoderDecoderModel
2from PIL import Image
3
4processor = TrOCRProcessor.from_pretrained("YOUR_USERNAME/deepseek-ocr-german")
5model = VisionEncoderDecoderModel.from_pretrained("YOUR_USERNAME/deepseek-ocr-german")
6
7# Multiple images
8images = [Image.open(f"image_{i}.jpg").convert("RGB") for i in range(5)]
9
10# Batch process
11pixel_values = processor(images, return_tensors="pt", padding=True).pixel_values
12generated_ids = model.generate(pixel_values)
13generated_texts = processor.batch_decode(generated_ids, skip_special_tokens=True)
14
15for text in generated_texts:
16 print(text)1import torch
2from transformers import TrOCRProcessor, VisionEncoderDecoderModel
3from PIL import Image
4
5device = "cuda" if torch.cuda.is_available() else "cpu"
6
7processor = TrOCRProcessor.from_pretrained("YOUR_USERNAME/deepseek-ocr-german")
8model = VisionEncoderDecoderModel.from_pretrained("YOUR_USERNAME/deepseek-ocr-german").to(device)
9
10image = Image.open("german_text.jpg").convert("RGB")
11pixel_values = processor(image, return_tensors="pt").pixel_values.to(device)
12
13generated_ids = model.generate(pixel_values)
14text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
15print(text)1# Example training configuration
2from transformers import Seq2SeqTrainer, Seq2SeqTrainingArguments
3
4training_args = Seq2SeqTrainingArguments(
5 output_dir="./deepseek-ocr-german",
6 per_device_train_batch_size=8,
7 per_device_eval_batch_size=8,
8 learning_rate=5e-5,
9 num_train_epochs=10,
10 logging_steps=100,
11 save_steps=1000,
12 eval_steps=1000,
13 evaluation_strategy="steps",
14 save_total_limit=2,
15 fp16=True,
16 predict_with_generate=True,
17)1@misc{deepseek-ocr-german,
2 author = {Santosh Pandit},
3 title = {DeepSeek OCR - German Fine-tuned},
4 year = {2025},
5 publisher = {HuggingFace},
6 howpublished = {\url{https://huggingface.co/YOUR_USERNAME/deepseek-ocr-german}},
7}