File size: 470 Bytes
770f950
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from my_chess.scripts import HumanVsBot
from my_chess.learner.environments import Chess

from huggingface.pushmodels import DCMinMax


def main(kwargs=None):
    modelminmax = DCMinMax.from_pretrained("mzimm003/DeepChessReplicationMinMax")
    play = HumanVsBot(
        model=modelminmax,
        environment=Chess(render_mode="human"),
        extra_model_environment_context=lambda env: {"board":env.board}
    )
    play.run()

if __name__ == "__main__":
    main()