Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -7,9 +7,9 @@ from paddleocr import PaddleOCR
|
|
7 |
from doctr.io import DocumentFile
|
8 |
|
9 |
|
10 |
-
|
11 |
-
os.environ['
|
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)
|