Spaces:

napatswift
/

votecount-ml-be

Build error

napatswift commited on Sep 12, 2023

Commit

3c11248

•

1 Parent(s): 49aa0b6

add recog model

Files changed (1) hide show

main.py CHANGED Viewed

@@ -8,8 +8,11 @@ import torch
 print('Loading model...')
 device = 'gpu' if torch.cuda.is_available() else 'cpu'
-ocr = MMOCRInferencer(det='model/text-det/psenet.py',
             det_weights='model/text-det/psenet.pth',
             device=device)
 def get_rec(points):

 print('Loading model...')
 device = 'gpu' if torch.cuda.is_available() else 'cpu'
+ocr = MMOCRInferencer(
+            det='model/text-det/psenet.py',
             det_weights='model/text-det/psenet.pth',
+            rec='model/text-recog/config.py',
+            rec_weights='model/text-recog/model.pth',
             device=device)
 def get_rec(points):