PlayChessVsAI / app.py
mzimm003
Include chess models for pretrain loading.
0154f58
raw
history blame
467 Bytes
from my_chess.scripts.scripts import HumanVsBot
from my_chess.learner.environments import Chess
from chessmodels 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()