Changing device location

app.py

@@ -34,10 +34,10 @@ device = torch.device('cuda' if torch.cuda.is_available else 'cpu')
 
 # loading model
 PATH = 'pytorchBERTmodel'
-model = torch.load(PATH)
+model = torch.load(PATH, map_location=torch.device('cpu'))
 model.eval()
 
-model.to(device)
+model.to('cpu')
 
 p = pipeline("automatic-speech-recognition")