Update README.md
Browse files
README.md
CHANGED
|
@@ -65,8 +65,8 @@ You can play against the model using the provided `play.py` script.
|
|
| 65 |
from transformers import BertForSequenceClassification, PreTrainedTokenizerFast
|
| 66 |
import torch
|
| 67 |
|
| 68 |
-
model = BertForSequenceClassification.from_pretrained("
|
| 69 |
-
tokenizer = PreTrainedTokenizerFast.from_pretrained("
|
| 70 |
|
| 71 |
board = "X..O....." # X to move
|
| 72 |
inputs = tokenizer(board, return_tensors="pt")
|
|
|
|
| 65 |
from transformers import BertForSequenceClassification, PreTrainedTokenizerFast
|
| 66 |
import torch
|
| 67 |
|
| 68 |
+
model = BertForSequenceClassification.from_pretrained("abdelkader-dev/FemtoXO")
|
| 69 |
+
tokenizer = PreTrainedTokenizerFast.from_pretrained("abdelkader-dev/FemtoXO")
|
| 70 |
|
| 71 |
board = "X..O....." # X to move
|
| 72 |
inputs = tokenizer(board, return_tensors="pt")
|