Upload app.py

app.py

@@ -0,0 +1,21 @@
+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}")