map location cpu

app.py

@@ -25,7 +25,7 @@ model = BertForSequenceClassification.from_pretrained(model_name,
                                                       output_attentions=False,
                                                       output_hidden_states=False)
 
-torch_model = torch.load('FineTune_IndoLEM_BERT_H_Mean_Pooling_LR1E-5_BS2_epoch_9.model')
+torch_model = torch.load('FineTune_IndoLEM_BERT_H_Mean_Pooling_LR1E-5_BS2_epoch_9.model', map_location=torch.device('cpu'))
 torch_model['classifier.weight'] = torch_model.pop('out.weight')
 torch_model['classifier.bias'] = torch_model.pop('out.bias')
 model.load_state_dict(torch_model)