diff --git a/.dockerignore b/.dockerignore index b5868e2..183222c 100644 --- a/.dockerignore +++ b/.dockerignore @@ -13,6 +13,7 @@ Machine_Learning/dataset Machine_Learning/dataset.zip Machine_Learning/dataset_*.zip Machine_Learning/model +!Machine_Learning/model/model.onnx Machine_Learning/best_model/* !Machine_Learning/best_model/best_model.keras apps/web/dist diff --git a/apps/ml-service/Cargo.toml b/apps/ml-service/Cargo.toml index d3fe32c..66240f9 100644 --- a/apps/ml-service/Cargo.toml +++ b/apps/ml-service/Cargo.toml @@ -8,7 +8,7 @@ anyhow = "1.0" axum = { version = "0.7", features = ["multipart"] } image = "0.25" ndarray = "0.15" -ort = "2.0.0-rc.10" +ort = { version = "2.0.0-rc.10", features = ["download-binaries"] } serde = { version = "1.0", features = ["derive"] } serde_json = "1.0" tokio = { version = "1.0", features = ["macros", "rt-multi-thread", "net"] }