| """ | |
| Basic usage example for the Custom 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, use_native=True) | |
| print(f"Extracted text: {result['text']}") | |
| print(f"Confidence: {result['confidence']:.3f}") | |
| return result | |
| if __name__ == "__main__": | |
| basic_ocr_example() | |