Update README.md

README.md

@@ -1,3 +1,17 @@
+---
+datasets:
+- zefang-liu/phishing-email-dataset
+language:
+- en
+base_model:
+- google-bert/bert-base-uncased
+library_name: transformers
+tags:
+- phishing
+- email
+- detection
+- scam
+---
 # BERT Model for Phishing Detection
 
 This repository contains the fine-tuned **BERT model** for detecting phishing emails. The model has been trained to classify emails as either **phishing** or **legitimate** based on their body text.
@@ -50,21 +64,4 @@ This repository contains the fine-tuned **BERT model** for detecting phishing em
 
 4. **Expected Outputs**:
    1: Phishing
-   0: Legitimate
-
----
-datasets:
-- zefang-liu/phishing-email-dataset
----
-language:
-- en
----
-base_model:
-- google-bert/bert-base-uncased
----
-tags:
-- phishing
-- email
-- scam
-- detection
----
+   0: Legitimate