Update app.py

app.py

@@ -15,14 +15,14 @@ from paddleocr import PaddleOCR
 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'
+# 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)
+model = ocr_predictor(det_arch='db_resnet50', reco_arch='crnn_vgg16_bn', pretrained=True)