Update README.md

README.md

@@ -7,6 +7,10 @@ tags:
 model-index:
 - name: gpt-neo-125m-finetuned-shakespeare
   results: []
+datasets:
+- 2nji/Shakespeare_Corpus
+language:
+- en
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -18,17 +22,23 @@ This model is a fine-tuned version of [EleutherAI/gpt-neo-125m](https://huggingf
 It achieves the following results on the evaluation set:
 - Loss: 4.1126
 
-## Model description
+## How to use
 
-More information needed
+You can use this model directly with a pipeline for text generation. This example generates a different sequence each time it's run:
 
-## Intended uses & limitations
+```python
+from transformers import pipeline
 
-More information needed
+generator = pipeline('text-generation', model='2nji/gpt-neo-125m-finetuned-shakespeare')
+
+generator("And all that", do_sample=True, min_length=20)
+
+# [{'generated_text': "And all that in heaven is free: Thou bestow'd on God, so to my house, and in the"}]
+```
 
 ## Training and evaluation data
 
-More information needed
+This model was finetuned using the the [Shakespare_corpus](https://huggingface.co/datasets/2nji/Shakespeare_Corpus) Dataset
 
 ## Training procedure
 
@@ -57,4 +67,4 @@ The following hyperparameters were used during training:
 - Transformers 4.48.3
 - Pytorch 2.5.1+cu124
 - Datasets 3.3.2
-- Tokenizers 0.21.0
+- Tokenizers 0.21.0