examples/basic_usage.py

"""
Basic usage example for the Custom PaliGemma OCR Model.
"""

from transformers import AutoModel
from PIL import Image

def basic_ocr_example():
    """Basic OCR usage example."""
    
    # Load model
    model = AutoModel.from_pretrained("your-username/your-model-name", trust_remote_code=True)
    
    # Load image
    image = Image.open("document.jpg")
    
    # Extract text
    result = model.generate_ocr_text(image)
    
    print(f"Extracted text: {result['text']}")
    print(f"Confidence: {result['confidence']:.3f}")
    print(f"Quality: {result['quality']}")
    print(f"Method: {result['method']}")
    
    return result

if __name__ == "__main__":
    basic_ocr_example()