Update app.py

app.py

@@ -17,6 +17,9 @@ from doctr.io import DocumentFile
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"
+# Load the doctr OCR model
+os.environ['USE_TORCH'] = 'YES'
+os.environ['USE_TF'] = 'NO'
 
 from doctr.models import ocr_predictor
 ocr_model = ocr_predictor(pretrained=True)