liuyizhang commited on
Commit
af76788
1 Parent(s): 9988241

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -347,8 +347,8 @@ config = dict(
347
  )
348
  config = Config(config)
349
 
350
- class Ram_Predictor(RamPredictor, device='cpu'):
351
- def __init__(self,config):
352
  self.config = config
353
  self.device = torch.device(device)
354
  self._build_model()
 
347
  )
348
  config = Config(config)
349
 
350
+ class Ram_Predictor(RamPredictor):
351
+ def __init__(self, config, device='cpu'):
352
  self.config = config
353
  self.device = torch.device(device)
354
  self._build_model()