KarthikeyanDev
/

ATS_RESUME_CHECKER

Text Classification

Model card Files Files and versions Community

ATS_RESUME_CHECKER / src /model.py

KarthikeyanDev's picture

Initial commit with model and scripts

f08d3c9 about 1 month ago

181 Bytes

	from transformers import AutoModelForSequenceClassification

	def load_model():
	return AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=2)