Views
No views yet
| اصل تصویر (Original Image) | ماڈل کی پہچان (OCR Prediction) |
|---|---|
 | سنا ہے لوگ اسے آنکھ بھر کے دیکھتے ہیں سو اس کے شہر میں کچھ دن ٹھہر کے دیکھتے ہیں |
 | سنا ہے ربط ہے اس کو خراب حالوں سے سو اپنے آپ کو برباد کر کے دیکھتے ہیں |
 | سنا ہے درد کی گاہک ہے چشم ناز اس کی سو ہم بھی اس کی گلی سے گزر کے دیکھتے ہیں |
 | سنا ہے اس کو بھی ہے شعر و شاعری سے شغف سو ہم بھی معجزے اپنے ہنر کے دیکھتے ہیں |
 | سنا ہے بولے تو باتوں سے پھول جھڑتے ہیں یہ بات ہے تو چلو بات کر کے دیکھتے ہیں |
1import torch
2from transformers import TrOCRProcessor, VisionEncoderDecoderModel
3from PIL import Image
4
5# ماڈل لوڈ کریں
6repo_id = "Khurram123/urdu-poetry-trocr"
7processor = TrOCRProcessor.from_pretrained(repo_id)
8model = VisionEncoderDecoderModel.from_pretrained(repo_id).to("cuda" if torch.cuda.is_available() else "cpu")
9
10def get_prediction(image_path):
11 image = Image.open(image_path).convert("RGB")
12 pixel_values = processor(image, return_tensors="pt").pixel_values.to(model.device)
13
14 generated_ids = model.generate(
15 pixel_values,
16 max_length=256,
17 num_beams=5,
18 repetition_penalty=3.0,
19 length_penalty=2.0,
20 early_stopping=False
21 )
22 return processor.batch_decode(generated_ids, skip_special_tokens=True)[0]