Update README.md

README.md

@@ -6,17 +6,21 @@ tags:
 
 ## Model description
 
-More information needed
+Implement a Transformer block as a Keras layer and use it for text classification.
 
-## Intended uses & limitations
+For details on the implementation, please see the original link on [keras](https://keras.io/examples/nlp/text_classification_with_transformer/)
 
-More information needed
+Full credits to: [Apoorv Nandan](https://twitter.com/NandanApoorv)
 
 ## Training and evaluation data
 
-More information needed
+The model is trained and evaluated on the IMDB dataset for sentiment analysis.
 
-## Training procedure
+Details on the dstaset can be found on [keras](https://keras.io/api/datasets/imdb/)
+
+Note that the keras dataset is already tokenized, so the model doesn't have an associated tokenizer. Since the hosted text classification pipeline only accepts text as input, it is currently broken.
+
+For a live demo of the model using my own tokenizer, please use the [space](https://huggingface.co/spaces/keras-io/text-classification-with-transformer)
 
 ### Training hyperparameters