Spaces:

anhalu
/

transformer-ocr

Runtime error

anhalu commited on Jun 26, 2023

Commit

8183312

•

1 Parent(s): 1bbb4b6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from vietocr.tool.config import Cfg
 config = Cfg.load_config_from_name('vgg_transformer')
 config['weights'] = 'ocr.pth'
-config['device'] = 'cuda:0'
 config['cnn']['pretrained']=False
 config['predictor']['beamsearch']=False

 config = Cfg.load_config_from_name('vgg_transformer')
 config['weights'] = 'ocr.pth'
+config['device'] = 'cpu'
 config['cnn']['pretrained']=False
 config['predictor']['beamsearch']=False