Update README.md

README.md

@@ -1,3 +1,16 @@
+---
+license: apache-2.0
+datasets:
+- stanfordnlp/imdb
+language:
+- en
+metrics:
+- accuracy
+tags:
+- sentiment-analysis
+- distilbert
+- lora
+---
 # distilbert-base-uncased-imdb-lora
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on the [IMDb dataset](https://huggingface.co/datasets/imdb) for sentiment analysis, using LoRA (Low-Rank Adaptation) for efficient fine-tuning.
 
@@ -70,5 +83,4 @@ print(map_labels(classifier("The movie is bad")))   # Output: [{'label': 'NEGATI
 ## Limitations
 - Fine-tuned on IMDb reviews; may not generalize to other domains.
 - Confidence scores (~0.56 for positive) are moderate; further tuning can improve them.
-- Only supports binary classification (positive/negative).
-
+- Only supports binary classification (positive/negative).