Update app.py

app.py

@@ -6,7 +6,9 @@ from transformers import BertTokenizer, BertForSequenceClassification
 
 
 tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
-model.load_state_dict(torch.load('/content/Fake-news-detection-bert-based-uncased/model_after_train.pt', map_location=torch.device('cpu')), strict=False)
+model = BertForSequenceClassification.from_pretrained("bert-base-uncased")
+
+model.load_state_dict(torch.load('model_after_train.pt', map_location=torch.device('cpu')), strict=False)
 model.eval()
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')