MohamedRashad/arabic-small-nougat

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

1. Text Hallucination: The model may occasionally generate repeated or incorrect text due to the inherent complexities of OCR tasks.
2. Erroneous Image Paths: There are instances where the model outputs image paths that are not relevant to the input, indicating occasional confusion.
3. 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.