Update app.py

app.py

@@ -6,15 +6,16 @@ import torch
 from transformers import BertForSequenceClassification, BertTokenizer
 import gradio as gr
 
-# Load your BERT model
 # Load the model architecture
 model = BertForSequenceClassification.from_pretrained('bert-base-uncased', num_labels=2)
 
-# Load the state dict without weights_only
+# Load the state dict, ensuring tensors are on the CPU
 try:
     model.load_state_dict(torch.load('bert_model_complete.pth', map_location=torch.device('cpu')), strict=False)
 except Exception as e:
     print(f"Error loading state dict: {e}")
+
+    
 model.eval()  # Set the model to evaluation mode