Update README.md

README.md

@@ -1,24 +1,24 @@
----
-library_name: peft
-license: apache-2.0
-base_model: distilbert-base-uncased
-tags:
-- base_model:adapter:distilbert-base-uncased
-- lora
-- transformers
-metrics:
-- accuracy
-model-index:
-- name: distilbert-base-uncased-lora-text-classification
-  results: []
----
+---
+library_name: peft
+license: apache-2.0
+base_model: distilbert-base-uncased
+tags:
+- base_model:adapter:distilbert-base-uncased
+- lora
+- transformers
+metrics:
+- accuracy
+model-index:
+- name: distilbert-base-uncased-lora-text-classification
+  results: []
+---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
 # distilbert-base-uncased-lora-text-classification
 
-This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the IMDb dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.9717
 - Accuracy: {'accuracy': 0.872}