fix cpu setting

app.py

@@ -13,7 +13,7 @@ model = BertForSequenceClassification.from_pretrained(model_name,
                                                       num_labels=len(label_dict),
                                                       output_attentions=False,
                                                       output_hidden_states=False)
-model.load_state_dict(torch.load('finetuned_BERT_epoch_9.model'), map_location=torch.device('cpu'))
+model.load_state_dict(torch.load('finetuned_BERT_epoch_9.model', map_location=torch.device('cpu')))
 
 from transformers import TextClassificationPipeline
 pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=True)