Spaces:

Genzo1010
/

ocr_zero

Runtime error

Genzo1010 commited on Sep 17, 2024

Commit

0d43dab

verified ·

1 Parent(s): d44e780

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ app = FastAPI()
 # Load the doctr OCR model
 os.environ['USE_TORCH'] = 'YES'
 os.environ['USE_TF'] = 'NO'
-ocr_model = ocr_predictor(det_arch='db_resnet50', reco_arch='crnn_vgg16_bn', pretrained=True)
 def ocr_with_doctr(file):
     text_output = ''

 # Load the doctr OCR model
 os.environ['USE_TORCH'] = 'YES'
 os.environ['USE_TF'] = 'NO'
+ocr_model = ocr_predictor(pretrained=True)
 def ocr_with_doctr(file):
     text_output = ''