Spaces:

napatswift
/

votecount-ml-be

Build error

napatswift commited on May 14, 2023

Commit

28d6a1a

1 Parent(s): aa41622

Fix bug

Files changed (1) hide show

main.py CHANGED Viewed

@@ -7,8 +7,8 @@ import torch
 print('Loading model...')
 device = 'gpu' if torch.cuda.is_available() else 'cpu'
-table_det = init_detector('model/table-det/config.py',
-                          'model/table-det/model.pth', device=device)
 ocr = MMOCRInferencer(det='model/det/config.py',
             det_weights='model/det/model.pth',

 print('Loading model...')
 device = 'gpu' if torch.cuda.is_available() else 'cpu'
+# table_det = init_detector('model/table-det/config.py',
+#                           'model/table-det/model.pth', device=device)
 ocr = MMOCRInferencer(det='model/det/config.py',
             det_weights='model/det/model.pth',