Update app.py

app.py

@@ -9,7 +9,9 @@ import io
 app = FastAPI()
 
 # Load the doctr OCR model
-ocr_model = ocr_predictor(pretrained=True)
+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).to(device)
 
 def ocr_with_doctr(file):
     text_output = ''