Genzo1010 commited on
Commit
1f47d3f
·
verified ·
1 Parent(s): 8ae607e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -7,9 +7,9 @@ from paddleocr import PaddleOCR
7
  from doctr.io import DocumentFile
8
 
9
 
10
- import os
11
- os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
12
-
13
 
14
  from doctr.models import ocr_predictor
15
  model = ocr_predictor(det_arch='db_resnet50', reco_arch='crnn_vgg16_bn', pretrained=True)
 
7
  from doctr.io import DocumentFile
8
 
9
 
10
+ ]# Set environment variable for PyTorch usage
11
+ os.environ['USE_TF'] = '0' # Set TensorFlow to off
12
+ os.environ['USE_TORCH'] = '1' # Set PyTorch to on
13
 
14
  from doctr.models import ocr_predictor
15
  model = ocr_predictor(det_arch='db_resnet50', reco_arch='crnn_vgg16_bn', pretrained=True)