Spaces:
Sleeping
Sleeping
Update pdf/utils/tt_module.py
Browse files- pdf/utils/tt_module.py +1 -1
pdf/utils/tt_module.py
CHANGED
|
@@ -13,7 +13,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 13 |
model.to(device)
|
| 14 |
structure_model = TableTransformerForObjectDetection.from_pretrained("microsoft/table-structure-recognition-v1.1-all")
|
| 15 |
structure_model.to(device)
|
| 16 |
-
reader = easyocr.Reader(['en'])
|
| 17 |
|
| 18 |
def pdf_to_img(pdf_path):
|
| 19 |
image_list = []
|
|
|
|
| 13 |
model.to(device)
|
| 14 |
structure_model = TableTransformerForObjectDetection.from_pretrained("microsoft/table-structure-recognition-v1.1-all")
|
| 15 |
structure_model.to(device)
|
| 16 |
+
reader = easyocr.Reader(['en'], gpu=False)
|
| 17 |
|
| 18 |
def pdf_to_img(pdf_path):
|
| 19 |
image_list = []
|