Arabic Small Nougat
End-to-End Structured OCR for Arabic books.
Description
Github π€ Hugging Face π Paper ποΈ Data π½οΈ Demo
The arabic-small-nougat OCR is an end-to-end structured Optical Character Recognition (OCR) system designed specifically for the Arabic language.
The model is based on the facebook/nougat-small architecture and has been fine-tuned using the Khatt dataset along with a custom dataset created for this purpose.
How to Get Started with the Model
Demo: https://huggingface.co/spaces/MohamedRashad/Arabic-Nougat
Or, use the code below to get started with the model locally.
from PIL import Image
import torch
from transformers import NougatProcessor, VisionEncoderDecoderModel
# Load the model and processor
processor = NougatProcessor.from_pretrained("MohamedRashad/arabic-small-nougat")
model = VisionEncoderDecoderModel.from_pretrained("MohamedRashad/arabic-small-nougat")
device = "cuda" if torch.cuda.is_available() else "cpu"
model.to(device)
context_length = 2048
def predict(img_path):
# prepare PDF image for the model
image = Image.open(img_path)
pixel_values = processor(image, return_tensors="pt").pixel_values
# generate transcription
outputs = model.generate(
pixel_values.to(device),
min_length=1,
max_new_tokens=context_length,
bad_words_ids=[[processor.tokenizer.unk_token_id]],
)
page_sequence = processor.batch_decode(outputs, skip_special_tokens=True)[0]
page_sequence = processor.post_process_generation(page_sequence, fix_markdown=False)
return page_sequence
print(predict("path/to/page_image.jpg"))
Bias, Risks, and Limitations
- Text Hallucination: The model may occasionally generate repeated or incorrect text due to the inherent complexities of OCR tasks.
- Erroneous Image Paths: There are instances where the model outputs image paths that are not relevant to the input, indicating occasional confusion.
- Context Length Constraint: The model has a maximum context length of 2048 tokens, which may result in incomplete transcriptions for longer book pages.
Intended Use
The arabic-small-nougat OCR is designed for tasks that involve converting images of Arabic book pages into structured text, especially when Markdown format is desired. It is suitable for applications in the field of digitizing Arabic literature and facilitating text extraction from printed materials.
Ethical Considerations
It is crucial to be aware of the model's limitations, particularly in instances where accurate OCR results are critical. Users are advised to verify and review the output, especially in scenarios where precision is paramount.
Model Details
- Developed by: Mohamed Rashad
- Model type: VisionEncoderDecoderModel
- Language(s) (NLP): Arabic & English
- License: GPL 3.0
- Finetuned from model: nougat-small
Acknowledgment
If you use or build upon the Arabic Small Nougat OCR, please acknowledge the model developer and the open-source community for their contributions. Additionally, be sure to include a copy of the GPL 3.0 license with any redistributed or modified versions of the model.
By selecting the GPL 3.0 license, you promote the principles of open source and ensure that the benefits of the model are shared with the broader community.
Citation
If you find this model useful, please consider citing the original facebook/nougat-small model and the datasets used for fine-tuning, including the Khatt dataset and any details regarding the custom dataset.
@misc{rashad2024arabicnougatfinetuningvisiontransformers,
title={Arabic-Nougat: Fine-Tuning Vision Transformers for Arabic OCR and Markdown Extraction},
author={Mohamed Rashad},
year={2024},
eprint={2411.17835},
archivePrefix={arXiv},
primaryClass={cs.CL},
url={https://arxiv.org/abs/2411.17835},
}
@misc {mohamed_rashad_2024,
author = { {Mohamed Rashad} },
title = { arabic-small-nougat (Revision 48741d4) },
year = 2024,
url = { https://huggingface.co/MohamedRashad/arabic-small-nougat },
doi = { 10.57967/hf/3534 },
publisher = { Hugging Face }
}
Disclaimer
The arabic-small-nougat OCR is a tool provided "as is," and the developers make no guarantees regarding its suitability for specific tasks. Users are encouraged to thoroughly evaluate the model's output for their particular use cases and requirements.
- Downloads last month
- 6,523
Model tree for MohamedRashad/arabic-small-nougat
Base model
facebook/nougat-small