test_ocr_1

Sleeping

B2W1234 commited on Jun 22

Commit

17fbc0d

•

1 Parent(s): 51d089d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -103,7 +103,7 @@ if args.refine:
 config = Cfg.load_config_from_name('vgg_transformer')
 config['export'] = 'transformerocr_checkpoint.pth'
-config['device'] = 'cuda'
 config['predictor']['beamsearch'] = False
 detector = Predictor(config)

 config = Cfg.load_config_from_name('vgg_transformer')
 config['export'] = 'transformerocr_checkpoint.pth'
+config['device'] = 'cpu'
 config['predictor']['beamsearch'] = False
 detector = Predictor(config)