Update README.md

README.md

@@ -5,26 +5,22 @@ tags:
 model-index:
 - name: chatbot
   results: []
+datasets:
+- Anthropic/hh-rlhf
+library_name: transformers
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# chatbot
+# Chatbot
 
-This model is a fine-tuned version of [EleutherAI/gpt-neo-125M](https://huggingface.co/EleutherAI/gpt-neo-125M) on an unknown dataset.
+This model is a fine-tuned version of [EleutherAI/gpt-neo-125M](https://huggingface.co/EleutherAI/gpt-neo-125M) on an Anthropic's HH-RLHF dataset.
 
-## Model description
-
-More information needed
 
 ## Intended uses & limitations
 
-More information needed
-
-## Training and evaluation data
-
-More information needed
+This model is designed for research purposes only. I highly suggest not to use this in production.
 
 ## Training procedure
 
@@ -40,9 +36,12 @@ The following hyperparameters were used during training:
 - lr_scheduler_warmup_steps: 500
 - num_epochs: 2
 
+### Tensorboard
+You can view data about this model in [this](https://tensorboard.dev/experiment/lgu6VRXXR1C9lndVzbl3Ng]) tensorboard.
+
 ### Framework versions
 
 - Transformers 4.26.1
 - Pytorch 1.13.1+cu116
 - Datasets 2.10.1
-- Tokenizers 0.13.2
+- Tokenizers 0.13.2