Update README.md

README.md

@@ -8,6 +8,14 @@ metrics:
 model-index:
 - name: GPT2-small-LoRA-finetuned-amazon-reviews-en-classification
   results: []
+datasets:
+- mteb/amazon_reviews_multi
+language:
+- en
+widget:
+- text: It`s an amazing product
+- text: I hate this product
+- text: It's ok, but a bit expensive
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,22 +23,25 @@ should probably proofread and complete it, then remove this comment. -->
 
 # GPT2-small-LoRA-finetuned-amazon-reviews-en-classification
 
-This model is a fine-tuned version of [openai-community/gpt2](https://huggingface.co/openai-community/gpt2) on an unknown dataset.
+This model is a fine-tuned version of [openai-community/gpt2](https://huggingface.co/openai-community/gpt2) on [mteb/amazon_reviews_multi](https://huggingface.co/datasets/mteb/amazon_reviews_multi) dataset.
+
+It is the result of the post [Fine tunning SML](https://maximofn.com/lora/)
+
 It achieves the following results on the evaluation set:
 - Loss: 1.5203
 - Accuracy: 0.3374
 
 ## Model description
 
-More information needed
+This model provides classification of reviews in english
 
 ## Intended uses & limitations
 
-More information needed
+Classifiction of reviews in english
 
 ## Training and evaluation data
 
-More information needed
+It is training on [mteb/amazon_reviews_multi](https://huggingface.co/datasets/mteb/amazon_reviews_multi) dataset
 
 ## Training procedure
 
@@ -61,4 +72,4 @@ The following hyperparameters were used during training:
 - Transformers 4.42.4
 - Pytorch 2.3.1+cu121
 - Datasets 2.20.0
-- Tokenizers 0.19.1
+- Tokenizers 0.19.1