| """ | |
| Constants for the demo. | |
| """ | |
| import os | |
| ONNX_MODEL_DIRECTORY = "demo/onnx-models" | |
| LEELA_MODEL_DIRECTORY = "demo/leela-models" | |
| FIGURE_DIRECTORY = "demo/figures" | |
| ONNX_MODEL_NAMES = [ | |
| f for f in os.listdir(ONNX_MODEL_DIRECTORY) | |
| if f.endswith(".onnx") | |
| ] | |
| LEELA_MODEL_NAMES = [ | |
| f for f in os.listdir(LEELA_MODEL_DIRECTORY) | |
| if f.endswith(".pb.gz") | |
| ] | |