OCR-image-to-text

Sleeping

Genzo1010 commited on Sep 18, 2024

Commit

7f814c9

verified ·

1 Parent(s): 124ee3d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ os.environ['USE_TF'] = '0'  # Set TensorFlow to off
 os.environ['USE_TORCH'] = '1'  # Set PyTorch to on
 from doctr.models import ocr_predictor
-model = ocr_predictor(det_arch='db_resnet50', reco_arch='crnn_vgg16_bn', pretrained=True)

 os.environ['USE_TORCH'] = '1'  # Set PyTorch to on
 from doctr.models import ocr_predictor
+ocr_model = ocr_predictor(det_arch='db_resnet50', reco_arch='crnn_vgg16_bn', pretrained=True)