Update app.py

app.py

@@ -16,10 +16,10 @@ tokenizer = BertTokenizer.from_pretrained('bert-base-uncased',
                                           do_lower_case=True)
 # Create BERT model
 model = BertForSequenceClassification.from_pretrained("bert-base-uncased",
-                                                      num_labels=len(label_dict),
+                                                      num_labels=2,
                                                       output_attentions=False,
                                                       output_hidden_states=False)
-model.load_state_dict(torch.load('/content/finetuned_BERT_epoch_10.model', map_location=torch.device('cpu')))
+model.load_state_dict(torch.load(f='finetuned_BERT_epoch_10.model', map_location=torch.device('cpu')))
 ### 3. Predict function ###
 
 # Create predict function