Update README.md

README.md

@@ -13,6 +13,8 @@ metrics:
 model-index:
 - name: GPT2-genre-detection
   results: []
+library_name: transformers
+pipeline_tag: text-classification
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -26,19 +28,24 @@ It achieves the following results on the evaluation set:
 - Accuracy: 0.4593
 - Matthews Correlation: 0.1010
 
-## Model description
+##  description
+
+[Data-Driven Science](https://huggingface.co/datadrivenscience) organized a competition where in the goal was to fine tune a model that can predict the genre of a movie from a given synopsis. There were a total of 10 genres as follows:
+```json
+{
+    "0": "horror",
+    "1": "adventure",
+    "2": "action",
+    "3": "crime",
+    "4": "mystery",
+    "5": "family",
+    "6": "scifi",
+    "7": "thriller",
+    "8": "fantasy",
+    "9": "romance"
+  }
+```
 
-More information needed
-
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
-
-More information needed
-
-## Training procedure
 
 ### Training hyperparameters
 
@@ -65,4 +72,4 @@ The following hyperparameters were used during training:
 - Transformers 4.36.2
 - Pytorch 2.1.0+cu121
 - Datasets 2.16.0
-- Tokenizers 0.15.0
+- Tokenizers 0.15.0