Update main.py

main.py

@@ -3,6 +3,9 @@ from PIL import ImageDraw, ImageFont, Image
 from transformers import AutoModelForTokenClassification, AutoProcessor
 import fitz  # PyMuPDF
 import io
+import os
+
+MODEL_KEY = os.getenv(MODEL_KEY)
 
 
 def extract_data_from_pdf(pdf_path, page_number=0):
@@ -80,7 +83,8 @@ def interference(example, page_number=0):
 
     # Process the image and words
     model = AutoModelForTokenClassification.from_pretrained(
-        "karida/LayoutLMv3_RFP"
+        "karida/LayoutLMv3_RFP",
+        use_auth_token=MODEL_KEY
     )
     processor = AutoProcessor.from_pretrained(
         "microsoft/layoutlmv3-base", apply_ocr=False