Neuracraft / model_loading.py
Erfan11's picture
Update model_loading.py
8366538 verified
raw
history blame
415 Bytes
from transformers import TFBertForSequenceClassification, BertTokenizerFast
# 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 = BertTokenizerFast.from_pretrained(model_name, use_auth_token="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")