Update README.md

README.md

@@ -9,6 +9,15 @@ tags:
 model-index:
 - name: GPT2-small-QLoRA-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
+pipeline_tag: text-classification
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -16,14 +25,17 @@ should probably proofread and complete it, then remove this comment. -->
 
 # GPT2-small-QLoRA-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/qlora/)
+
 It achieves the following results on the evaluation set:
 - Loss: 0.8883
 - Accuracy: 0.615
 
 ## Model description
 
-More information needed
+This model provides classification of reviews in english
 
 ## Intended uses & limitations