Spaces:

alinikkhah
/

SpoilerDetectionClassification

Runtime error

alinikkhah commited on Sep 27, 2024

Commit

52dcfd7

verified ·

1 Parent(s): d289606

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import BertForSequenceClassification
 # Load your BERT model
 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
 from transformers import BertTokenizer

 # Load your BERT model
 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'), weights_only=True), strict=False)
 model.eval()  # Set the model to evaluation mode
 import gradio as gr
 from transformers import BertTokenizer