bettermint commited on
Commit
dfac5b8
·
verified ·
1 Parent(s): 5118a20

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -8,7 +8,7 @@ import os
8
  from threading import Thread
9
  from queue import Queue, Empty
10
 
11
- os.environ['LD_LIBRARY_PATH'] = '/engines/maia:' + os.environ.get('LD_LIBRARY_PATH', '')
12
 
13
  def enqueue_output(out, queue):
14
  for line in iter(out.readline, b''):
@@ -82,7 +82,7 @@ app.add_middleware(
82
  async def websocket_endpoint(websocket: WebSocket, version: str):
83
  await websocket.accept()
84
 
85
- stockfish = EngineChess([f"engines/stockfish/stockfish-{version}-uci"])
86
 
87
  async def read_from_socket(websocket: WebSocket):
88
  async for data in websocket.iter_text():
@@ -104,7 +104,7 @@ async def websocket_endpoint(websocket: WebSocket, version: str):
104
  async def websocket_endpoint(websocket: WebSocket, elo: str):
105
  await websocket.accept()
106
 
107
- stockfish = EngineChess(["./engines/maia/lc0", "--backend=trivial", f"--weights=./engines/maia-{elo}.pb.gz"])
108
 
109
  async def read_from_socket(websocket: WebSocket):
110
  async for data in websocket.iter_text():
 
8
  from threading import Thread
9
  from queue import Queue, Empty
10
 
11
+ os.environ['LD_LIBRARY_PATH'] = '/engines/maia/lib:' + os.environ.get('LD_LIBRARY_PATH', '')
12
 
13
  def enqueue_output(out, queue):
14
  for line in iter(out.readline, b''):
 
82
  async def websocket_endpoint(websocket: WebSocket, version: str):
83
  await websocket.accept()
84
 
85
+ stockfish = EngineChess([f"./engines/stockfish/stockfish-{version}-uci"])
86
 
87
  async def read_from_socket(websocket: WebSocket):
88
  async for data in websocket.iter_text():
 
104
  async def websocket_endpoint(websocket: WebSocket, elo: str):
105
  await websocket.accept()
106
 
107
+ stockfish = EngineChess([f"./engines/maia/maia-{elo}/lc0", "--backend=trivial"])
108
 
109
  async def read_from_socket(websocket: WebSocket):
110
  async for data in websocket.iter_text():