| import os | |
| from transformers import LayoutLMv3Processor, AutoModel | |
| # Step 1: Create the cache directory | |
| cache_directory = os.path.expanduser('~/huggingface_cache') # Change path as needed | |
| os.makedirs(cache_directory, exist_ok=True) | |
| # Step 2: Set the HF_HOME environment variable | |
| os.environ['HF_HOME'] = cache_directory | |
| # Step 3: Load the processor and model from Hugging Face | |
| try: | |
| # Load the LayoutLMv3 processor | |
| processor = LayoutLMv3Processor.from_pretrained("HURIDOCS/pdf-document-layout-analysis") | |
| # Load the model | |
| model = AutoModel.from_pretrained("HURIDOCS/pdf-document-layout-analysis") | |
| print("Model and processor loaded successfully.") | |
| except Exception as e: | |
| print(f"An error occurred: {e}") | |