Update Predict.py

Predict.py

@@ -12,7 +12,7 @@ with open('answer.json', 'r', encoding = 'utf8') as f:
 swap_space = {v : k for k, v in answer_space.items()}
 
 
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+device = torch.device('cpu')
 model = BaseModel().to(device)
 model.load_state_dict(checkpoint['model_state_dict'], map_locaption = device)