Textwizai / load_model.py
Erfan11's picture
Update load_model.py
8626faa verified
raw
history blame
257 Bytes
from transformers import TFBertForSequenceClassification
import tensorflow as tf
# Load the TensorFlow model
model = TFBertForSequenceClassification.from_pretrained("Erfan11/Neuracraft", from_tf=True, use_auth_token="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")