Neuracraft / load_model.py
Erfan11's picture
Update load_model.py
58d1031 verified
raw
history blame
268 Bytes
from transformers import TFBertForSequenceClassification
# 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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")