from unsloth import FastVisionModel
import torch
from transformers import AutoModel
import os

os.environ["UNSLOTH_WARN_UNINITIALIZED"] = '0'

# Load the model
model, tokenizer = FastVisionModel.from_pretrained(
    "karankulshrestha/formula_ocr",
    load_in_4bit = False,
    auto_model = AutoModel,
    trust_remote_code = True,
    unsloth_force_compile = True,
    use_gradient_checkpointing = "unsloth",
)

# Enable fast inference
FastVisionModel.for_inference(model)

# Run OCR on an image
result = model.infer(
    tokenizer,
    prompt="\nFree OCR.",
    image_file="path/to/your/image.jpg",
    output_path="./output",
    base_size=1024,
    image_size=640,
    crop_mode=True,
    save_results=True,
)

print("OCR Result:", result)

Output

Citation

bibtex

@misc{formula_ocr},
  author = {Karan Kulshrestha},
  title = {Hand-written OCR Model},
  year = {2025},
  publisher = {HuggingFace},
  url = {https://huggingface.co/karankulshrestha/formula_ocr}
}