Update app.py

app.py

@@ -3,7 +3,11 @@ from PIL import Image
 import torch
 from torchvision.transforms import functional as F
 from transformers import TrOCRProcessor, VisionEncoderDecoderModel
+from transformers import AutoModel
 
+access_token = os.environ.get("HF_TOKEN")
+
+model = AutoModel.from_pretrained("trocrnew.pth", token=access_token)
 app = Flask(__name__)
 
 # Load the trained model and processor