Spaces:
Running
Running
Update main.py
Browse files
main.py
CHANGED
|
@@ -8,6 +8,8 @@ import os
|
|
| 8 |
from threading import Thread
|
| 9 |
from queue import Queue, Empty
|
| 10 |
|
|
|
|
|
|
|
| 11 |
def enqueue_output(out, queue):
|
| 12 |
for line in iter(out.readline, b''):
|
| 13 |
queue.put(line)
|
|
@@ -103,7 +105,7 @@ async def websocket_endpoint(websocket: WebSocket, version: str):
|
|
| 103 |
async def websocket_endpoint(websocket: WebSocket, elo: str):
|
| 104 |
await websocket.accept()
|
| 105 |
|
| 106 |
-
stockfish = EngineChess(["
|
| 107 |
|
| 108 |
async def read_from_socket(websocket: WebSocket):
|
| 109 |
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:' + os.environ.get('LD_LIBRARY_PATH', '')
|
| 12 |
+
|
| 13 |
def enqueue_output(out, queue):
|
| 14 |
for line in iter(out.readline, b''):
|
| 15 |
queue.put(line)
|
|
|
|
| 105 |
async def websocket_endpoint(websocket: WebSocket, elo: str):
|
| 106 |
await websocket.accept()
|
| 107 |
|
| 108 |
+
stockfish = EngineChess(["./engines/maia/lc0", f"--weights=engines/maia-{elo}.pb.gz", "--backend=trivial"])
|
| 109 |
|
| 110 |
async def read_from_socket(websocket: WebSocket):
|
| 111 |
async for data in websocket.iter_text():
|