Update model_loading.py

model_loading.py

@@ -1,9 +1,8 @@
-# Use a pipeline as a high-level helper
-from transformers import pipeline
+from transformers import TFBertForSequenceClassification, BertTokenizerFast
 
-pipe = pipeline("text-generation", model="openai-community/gpt2")
-# Load model directly
-from transformers import AutoTokenizer, AutoModelForCausalLM
+# Define the model name
+model_name = "Erfan11/Neuracraft"
 
-tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
-model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")
+# Load the TensorFlow model and tokenizer
+model = TFBertForSequenceClassification.from_pretrained(model_name, use_auth_token="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")
+tokenizer = BertTokenizerFast.from_pretrained(model_name, use_auth_token="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")