Textwizai / load_model.py
Erfan11's picture
Update load_model.py
14a6661 verified
raw
history blame
407 Bytes
from transformers import TFBertForSequenceClassification, BertTokenizer
# 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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")