Spaces:

tensorlake
/

document-extractors

Sleeping

rishiraj commited on Jun 7

Commit

020474e

•

1 Parent(s): 7bf94d1

Update pdf/utils/tt_module.py

Files changed (1) hide show

pdf/utils/tt_module.py CHANGED Viewed

@@ -13,7 +13,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 structure_model = TableTransformerForObjectDetection.from_pretrained("microsoft/table-structure-recognition-v1.1-all")
 structure_model.to(device)
-reader = easyocr.Reader(['en'])
 def pdf_to_img(pdf_path):
     image_list = []

 model.to(device)
 structure_model = TableTransformerForObjectDetection.from_pretrained("microsoft/table-structure-recognition-v1.1-all")
 structure_model.to(device)
+reader = easyocr.Reader(['en'], gpu=False)
 def pdf_to_img(pdf_path):
     image_list = []