Update load_model.py

load_model.py

@@ -1,5 +1,8 @@
-from transformers import TFBertForSequenceClassification
-import tensorflow as tf
+from transformers import TFBertForSequenceClassification, BertTokenizer
 
-# Load the TensorFlow model
-model = TFBertForSequenceClassification.from_pretrained("Erfan11/Neuracraft", from_tf=True, use_auth_token="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")
+# Define the model name
+model_name = "Erfan11/Neuracraft"
+
+# Load the TensorFlow model and tokenizer
+model = TFBertForSequenceClassification.from_pretrained(model_name, use_auth_token="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")
+tokenizer = BertTokenizer.from_pretrained(model_name, use_auth_token="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")