Update README.md

README.md

@@ -1,3 +1,19 @@
+---
+language: en
+license: mit
+tags:
+- phishing
+- url-classification
+- bert
+datasets:
+- imanoop7/phishing_url_classification
+metrics:
+- accuracy
+- auc
+base_model:
+- google-bert/bert-base-uncased
+library_name: transformers
+---
 
 
 # BERT Phishing URL Detector
@@ -18,4 +34,4 @@ The model was trained on a custom dataset of URLs labeled as safe or unsafe. The
 
 ## Training procedure
 
-The model was fine-tuned using the Hugging Face Transformers library. Only the pooling layers were fine-tuned while the base BERT layers were frozen.
+The model was fine-tuned using the Hugging Face Transformers library. Only the pooling layers were fine-tuned while the base BERT layers were frozen.