Update README.md

README.md

@@ -1,3 +1,10 @@
+---
+license: apache-2.0
+tags:
+- text-classification
+- fine-tuning
+- resume classification
+---
 # DistilBERT Resume Classification Model
 
 This repository contains a fine-tuned DistilBERT model for classifying resume sentences into predefined categories. The model is trained on a dataset of resumes and can classify sentences into categories such as Personal Information, Experience, Summary, Education, Qualifications & Certificates, Skills, and Objectives.
@@ -31,6 +38,4 @@ from transformers import DistilBertTokenizerFast, DistilBertForSequenceClassific
 # Load the model and tokenizer
 model_name = "oussama120/Resume_Sentence_Classification"
 tokenizer = DistilBertTokenizerFast.from_pretrained(model_name)
-model = DistilBertForSequenceClassification.from_pretrained(model_name)
-
-
+model = DistilBertForSequenceClassification.from_pretrained(model_name)