Hugging Face's logo

Spaces:
Genzo1010
/
OCR-image-to-text
Sleeping

App Files Community
Genzo1010 commited on
Commit
dcc2b56
·
verified ·
1 Parent(s): 93a77dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -15,7 +15,9 @@ from paddleocr import PaddleOCR
15
  from doctr.io import DocumentFile
16
 
17
 
18
- os.environ["USE_TORCH"] = "1"
 
 
19
  from doctr.models import ocr_predictor
20
  ocr_model = ocr_predictor(pretrained=True)
21
 
 
15
  from doctr.io import DocumentFile
16
 
17
 
18
+ os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
19
+ os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"
20
+
21
  from doctr.models import ocr_predictor
22
  ocr_model = ocr_predictor(pretrained=True)
23