Update README.md

README.md

@@ -7,7 +7,8 @@ license: mit
 An generative AI made using [microsoft/DialoGPT-small](https://huggingface.co/microsoft/DialoGPT-small).
 Trained on:-
 https://www.kaggle.com/Cornell-University/movie-dialog-corpus
-It will next be trained on reddit data and maybe discord.
+https://www.kaggle.com/jef1056/discord-data
+
 Example:-
 ```python
 from transformers import AutoTokenizer, AutoModelWithLMHead
@@ -25,7 +26,7 @@ for step in range(4):
     chat_history_ids = model.generate(
         bot_input_ids, max_length=200,
         pad_token_id=tokenizer.eos_token_id,  
-        no_repeat_ngram_size=3,       
+        no_repeat_ngram_size=4,       
         do_sample=True, 
         top_k=100, 
         top_p=0.7,