Update app.py

app.py

@@ -22,8 +22,9 @@ app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024  # 16MB max upload
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 
 # Load OCR model - Microsoft's Donut model
-processor = DonutProcessor.from_pretrained("microsoft/donut-base")
-model = VisionEncoderDecoderModel.from_pretrained("microsoft/donut-base")
+print("Loading OCR model...")
+processor = DonutProcessor.from_pretrained("microsoft/donut-base", cache_dir="/huggingface_cache")
+model = VisionEncoderDecoderModel.from_pretrained("microsoft/donut-base", cache_dir="/huggingface_cache")
 
 # Move model to GPU if available
 device = "cuda" if torch.cuda.is_available() else "cpu"