Views
No views yet
| Original Image | Model Transcription (Urdu) |
|---|---|
![]() | اے نالہ! اے فصل کشور نہ اندوستاں چو مقا ہے تیری پیغامی کو جھک کر آسماں |
![]() | تجھ میں کچھ پیدا نہیں دیرینہ روز ٹکے نظاں تو جواں ہے گردش شام و سحر کے درمیاں |
![]() | ایک جلوہ تھا کلیم طور سینا کے لیے تو حج یہ ہے سراپا چشم پیما کے لیے |
![]() | امتحان دیدئہ ظاہر میں کو ہستاں بے تو پا سہاں اپنا ہے تو دیوار ہند ستاں بے تو |
![]() | مطلع ہوال فلک جس کا ہو وہ یواں ہے تو سوئے خلوت گاہ دل دامن کش انساں ہے تو |
![]() | چو نہال تیری ثریا سے ہیں سر گرم سخن تو نرمیں پرور پہنائے فلک تیرا وطن |
1from transformers import TrOCRProcessor, VisionEncoderDecoderModel
2from PIL import Image
3import torch
4
5# Load the fine-tuned model
6processor = TrOCRProcessor.from_pretrained("Khurram123/urdu-poetry-trocr-iqbal")
7model = VisionEncoderDecoderModel.from_pretrained("Khurram123/urdu-poetry-trocr-iqbal")
8
9# Load image and prepare pixels
10image = Image.open("sample_poetry.jpg").convert("RGB")
11pixel_values = processor(image, return_tensors="pt").pixel_values
12
13# Optimized generation parameters for Urdu Nastaliq
14generated_ids = model.generate(
15 pixel_values,
16 max_length=128,
17 num_beams=7, # Higher beams for complex ligature search
18 repetition_penalty=3.0, # Prevents character looping
19 length_penalty=1.5, # Encourages completion of full poetic lines
20 early_stopping=False
21)
22
23# Decode output
24transcription = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
25print(f"OCR Result: {transcription}")