Textwizai / load_model.py
Erfan11's picture
Create load_model.py
06f46aa verified
raw
history blame
No virus
196 Bytes
from transformers import BertForSequenceClassification
# Load the model from your Hugging Face repository
model = BertForSequenceClassification.from_pretrained("Erfan11/Neuracraft", from_tf=True)