Update app.py

app.py

@@ -2,7 +2,7 @@ import torch
 from transformers import BertForSequenceClassification
 
 # Load your BERT model
-model = BertForSequenceClassification.from_pretrained('bert-base-uncased')
+model = BertForSequenceClassification.from_pretrained('bert-base-uncased', num_labels=2)
 model.load_state_dict(torch.load('bert_model_complete.pth', map_location=torch.device('cpu')))
 model.eval()  # Set the model to evaluation mode
 import gradio as gr