Spaces:
Runtime error
Runtime error
File size: 618 Bytes
9742e85 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from transformers import AutoModelForSequenceClassification,AutoTokenizer
#***********************************************************************************************************
###FOR SENTIMENT CLASSIFICATION
chkpt='distilbert-base-uncased-finetuned-sst-2-english'
model=AutoModelForSequenceClassification.from_pretrained(chkpt)
tokenizer=AutoTokenizer.from_pretrained(chkpt)
#save the pertrained model
model.save_pretrained('sentiment_classifier/')
tokenizer.save_pretrained('sentiment_classifier/')
#*********************************************************************************************************** |