Update app.py

app.py

@@ -107,8 +107,11 @@ div[data-testid="stDownloadButton"] > button:hover { background: #1A1A2E !import
 def load_model():
     import torch
     from transformers import TrOCRProcessor, VisionEncoderDecoderModel
-    MODEL_ID  = "rasmodev/Handwriting_trocr_model"
-    processor = TrOCRProcessor.from_pretrained(MODEL_ID)
+    MODEL_ID = "rasmodev/Handwriting_trocr_model"
+    BASE_ID  = "microsoft/trocr-base-handwritten"
+    # Load processor from base model — has all required config files
+    # Load weights from fine-tuned model — contains trained parameters
+    processor = TrOCRProcessor.from_pretrained(BASE_ID)
     model     = VisionEncoderDecoderModel.from_pretrained(MODEL_ID)
     device    = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model.to(device).eval()