fixing bugs

app.py

@@ -41,7 +41,7 @@ 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(device))
 model.eval()
 
 model.to(device)