KarthikeyanDev
/

ATS_RESUME_CHECKER

Text Classification

Model card Files Files and versions

ATS_RESUME_CHECKER / src /model.py

KarthikeyanDev's picture

Initial commit with model and scripts

f08d3c9 6 months ago

history blame contribute delete

181 Bytes

	from transformers import AutoModelForSequenceClassification

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