Update README.md

README.md

@@ -11,13 +11,13 @@ This Hugging Face repository contains a fine-tuned DistilBERT model for sentimen
 ## Model Details
 
 - **Model Architecture**: Fine-tuned DistilBERT
-- **Sentiment Categories**: Positive, Negative, Neutral
+- **Sentiment Categories**: Neutral[0], Negative[1], Positive[2]
 - **Input Format**: Text-based clothing reviews
 - **Output Format**: Sentiment category labels
 
 ## Usage
 
-### Installation
+## Installation
 
 To use this model, you'll need to install the Hugging Face Transformers library and any additional dependencies.
 
@@ -25,6 +25,9 @@ To use this model, you'll need to install the Hugging Face Transformers library
 pip install transformers
 pip install torch
 
-## Model Loading
-You can easily load the pre-trained model for sentiment analysis using Hugging Face's AutoModelForSequenceClassification.
+## Usage
+You can easily load the pre-trained model for sentiment analysis using Hugging Face's DistilBertForSequenceClassification and DistilBertTokenizerFast.
+```bash
+pip install transformers
+pip install torch