mzimm003 commited on
Commit
5347397
·
1 Parent(s): 39de071

Fix bugs preventing bot from playing.

Browse files
Files changed (2) hide show
  1. Dockerfile +4 -4
  2. chessmodels.py +1 -1
Dockerfile CHANGED
@@ -21,7 +21,7 @@ COPY --chown=user . /app
21
 
22
  EXPOSE 8501
23
  CMD streamlit run app.py
24
- # --server.headless true \
25
- # --server.enableCORS false \
26
- # --server.enableXsrfProtection false \
27
- # --server.fileWatcherType none
 
21
 
22
  EXPOSE 8501
23
  CMD streamlit run app.py
24
+ --server.headless true \
25
+ --server.enableCORS false \
26
+ --server.enableXsrfProtection false \
27
+ --server.fileWatcherType none
chessmodels.py CHANGED
@@ -61,7 +61,7 @@ class DCMinMax(DeepChessAlphaBeta, PyTorchModelHubMixin):
61
  move_sort:Literal['none', 'random', 'evaluation'] = 'evaluation') -> None:
62
  input_sample = torch.rand(1,8,8,111)
63
  if board_evaluator is None:
64
- board_evaluator = DCFE.from_pretrained("mzimm003/DeepChessReplicationBoardEvaluator")
65
  super().__init__(input_sample, config=DeepChessAlphaBetaConfig(**dict(
66
  board_evaluator=board_evaluator,
67
  board_evaluator_config=board_evaluator_config,
 
61
  move_sort:Literal['none', 'random', 'evaluation'] = 'evaluation') -> None:
62
  input_sample = torch.rand(1,8,8,111)
63
  if board_evaluator is None:
64
+ board_evaluator = DCEvaluator.from_pretrained("mzimm003/DeepChessReplicationBoardEvaluator")
65
  super().__init__(input_sample, config=DeepChessAlphaBetaConfig(**dict(
66
  board_evaluator=board_evaluator,
67
  board_evaluator_config=board_evaluator_config,