diff --git a/README.md b/README.md index 16cacb7..3051f34 100644 --- a/README.md +++ b/README.md @@ -41,8 +41,14 @@ List of directories and files created by the project and its dependencies: The bot has several commands: ```text -► generate ⇢ Generates a message in the room. -► train ⇢ Trains the AI's mini GPT-2 model. +► speak ⇢ Generates a message in the room. +► prompt ⇢ Generate a message using a prompt. +``` + +To train the AI, run this command: + +``` +> python3 train.py ``` ### Setup