Upload folder using huggingface_hub

Dockerfile

@@ -41,8 +41,8 @@ RUN mkdir -p Neurex_Engine && \
     cp model.onnx Neurex_Engine/ || true && \
     cp vocab.json Neurex_Engine/ || true
 
-# Download the massive model_int8.onnx.data from our model repository directly into the root folder (where model_int8.onnx is)
-RUN curl -L -o ./model_int8.onnx.data https://huggingface.co/dpv007/Neurex-Weights/resolve/main/model_int8.onnx.data
+# Download the massive model.onnx.data from our model repository directly into the root folder (where model.onnx is)
+RUN curl -L -o ./model.onnx.data https://huggingface.co/dpv007/Neurex-Weights/resolve/main/model.onnx.data
 
 # Make engines executable
 RUN chmod +x Neurex_Engine/Neurex && \

app.py

@@ -119,7 +119,6 @@ async def make_move(req: MoveRequest):
                 print("ERROR: Rust engine is not loaded.")
                 return JSONResponse({'ai_move': None, 'eval': 0.0})
                 
-            # MCTS runs 150 simulations
             result = await rust_engine_pool.play(board, chess.engine.Limit(time=req.time_limit), info=chess.engine.INFO_ALL)
             ai_move = result.move.uci()
             

model.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b909959167c6467fa87bda3dbde035a6ff8fc98908785dd1268358b3d2cb0a93
-size 289721
+oid sha256:6a4c289f756799c983b4abb69849015416a8c03d4039eed21b0bb9c8e9c6e362
+size 110555

neural_engine/src/main.rs

@@ -482,11 +482,11 @@ fn main() {
     let num_threads = std::thread::available_parallelism().map(|n| n.get()).unwrap_or(4).min(8);
     let mut sessions: Vec<Option<Session>> = Vec::new();
     for _ in 0..num_threads {
-        let sess = if Path::new("model_int8.onnx").exists() {
+        let sess = if Path::new("model.onnx").exists() {
             Some(Session::builder().unwrap()
                 .with_optimization_level(GraphOptimizationLevel::Level3).unwrap()
                 .with_intra_threads(1).unwrap()
-                .commit_from_file("model_int8.onnx").unwrap())
+                .commit_from_file("model.onnx").unwrap())
         } else {
             None
         };