Update README.md

README.md

@@ -4,6 +4,8 @@ tags:
 model-index:
 - name: CodeGeneration-CodeT5-base
   results: []
+datasets:
+- ThoughtFocusAI/Python-CodeGeneration
 ---
 
 <!-- This model card has been generated automatically according to the information Keras had access to. You should
@@ -11,7 +13,7 @@ probably proofread and complete it, then remove this comment. -->
 
 # CodeGeneration-CodeT5-base
 
-This model was trained from the pretrained CodeT5-base model on an unknown dataset.
+This model was trained from the pretrained CodeT5-base model on [ThoughtFocusAI/Python-CodeGeneration](https://huggingface.co/datasets/ThoughtFocusAI/Python-CodeGeneration) dataset.
 It achieves the following results on the evaluation set:
 - ngram match: 0.014948515566713798 
 - weighted ngram match: 0.017803898905793227
@@ -33,7 +35,7 @@ Limited to simple queries.
 
 ## Training and evaluation data
 
-More information needed
+[ThoughtFocusAI/Python-CodeGeneration](https://huggingface.co/datasets/ThoughtFocusAI/Python-CodeGeneration)
 
 ## Training procedure
 
@@ -52,4 +54,4 @@ The following hyperparameters were used during training:
 - Transformers 4.29.2
 - TensorFlow 2.12.0
 - Datasets 2.12.0
-- Tokenizers 0.13.3
+- Tokenizers 0.13.3