Update load_model.py

load_model.py

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