Update README.md

README.md

@@ -8,8 +8,7 @@ language:
 # Model Card for Model ID
 
 <!-- Provide a quick summary of what the model is/does. -->
-
-
+This model classifies whether the text/email is spam or ham. 
 
 ## Model Details
 This model is a fine-tuned version of roberta-base using LoRA specifically for a binary classification task containing 
@@ -29,36 +28,16 @@ adaptations without altering the pre-trained weights directly.
 - **Language(s) (NLP):** [More Information Needed]
 - **Finetuned from model:** Roberta Base
 
-### Model Sources [optional]
-
-<!-- Provide the basic links for the model. -->
-
-- **Repository:** [More Information Needed]
-- **Paper [optional]:** [More Information Needed]
-- **Demo [optional]:** [More Information Needed]
-
-## Uses
-
-<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
+### Training Data
+The model was finetuned on truncated version of SetFit/enron_spam dataset. The SetFit/enron_spam consists of 33716 rows. 
+The truncated version consists of 1000 train samples and 1000 test samples with columns, text and label.
 
-### Direct Use
 
-<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
-
-[More Information Needed]
-
-
-### Recommendations
-
-<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
-
-Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
-
-## How to Get Started with the Model
+### Model Sources [optional]
 
-Use the code below to get started with the model.
+<!-- Provide the basic links for the model. -->
 
-[More Information Needed]
+- **Repository:** https://huggingface.co/FacebookAI/roberta-base
 
 ## Training Details