Update app.py

app.py

@@ -15,11 +15,9 @@ 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"] = "0"  # Use GPU 0, adjust if needed
+os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"
+
 
 from doctr.models import ocr_predictor
 model = ocr_predictor(det_arch='db_resnet50', reco_arch='crnn_vgg16_bn', pretrained=True)