hantech commited on
Commit
724755d
β€’
1 Parent(s): 295da9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ config = Cfg.load_config_from_name('vgg_transformer')
18
 
19
  config['cnn']['pretrained'] = True
20
  config['predictor']['beamsearch'] = True
21
- config['device'] = 'cuda:0' # mps
22
 
23
  recognitor = Predictor(config)
24
 
 
18
 
19
  config['cnn']['pretrained'] = True
20
  config['predictor']['beamsearch'] = True
21
+ config['device'] = 'cpucpu' # mps
22
 
23
  recognitor = Predictor(config)
24