RobPruzan commited on
Commit
7319f3e
1 Parent(s): 9c26cec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ PATH = 'pytorchBERTmodel'
44
  model = torch.load(PATH, map_location=torch.device('cpu'))
45
  model.eval()
46
 
47
- model.to(device)
48
 
49
  p = pipeline("automatic-speech-recognition")
50
 
 
44
  model = torch.load(PATH, map_location=torch.device('cpu'))
45
  model.eval()
46
 
47
+ model.to('cpu')
48
 
49
  p = pipeline("automatic-speech-recognition")
50