Update README.md

README.md

@@ -1,8 +1,8 @@
 ---
 license: mit
 datasets:
-- Xmaster6y/stockfish-debug
-name: Xmaster6y/gpt2-stockfish-debug
+- yp-edu/stockfish-debug
+name: yp-edu/gpt2-stockfish-debug
 results:
   - task: train
     metrics:
@@ -28,7 +28,7 @@ See my [blog post](https://yp-edu.github.io/projects/training-gpt2-on-stockfish-
 
 ## Training Details
 
-The model was trained during 1 epoch on the [Xmaster6y/stockfish-debug](https://huggingface.co/datasets/Xmaster6y/stockfish-debug) dataset (no hyperparameter tuning done). The samples are:
+The model was trained during 1 epoch on the [yp-edu/stockfish-debug](https://huggingface.co/datasets/yp-edu/stockfish-debug) dataset (no hyperparameter tuning done). The samples are:
 
 ```json
 {"prompt":"FEN: {fen}\nMOVE:", "completion": " {move}"}
@@ -66,8 +66,8 @@ def next_move(model, tokenizer, fen):
 
 
 board = chess.Board()
-model = AutoModelForCausalLM.from_pretrained("Xmaster6y/gpt2-stockfish-debug")
-tokenizer = AutoTokenizer.from_pretrained("Xmaster6y/gpt2-stockfish-debug")  # or "gpt2"
+model = AutoModelForCausalLM.from_pretrained("yp-edu/gpt2-stockfish-debug")
+tokenizer = AutoTokenizer.from_pretrained("yp-edu/gpt2-stockfish-debug")  # or "gpt2"
 tokenizer.pad_token = tokenizer.eos_token
 for i in range(100):
     fen = board.fen()