Update load_model.py

load_model.py

@@ -1,4 +1,5 @@
-from transformers import BertForSequenceClassification
+from transformers import TFBertForSequenceClassification
+import tensorflow as tf
 
-# Load the model from your Hugging Face repository
-model = BertForSequenceClassification.from_pretrained("Erfan11/Neuracraft", from_tf=True)
+# Load the TensorFlow model
+model = TFBertForSequenceClassification.from_pretrained("Erfan11/Neuracraft", from_tf=True, use_auth_token="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")