diff --git a/requirements_files/specific_golden_cpu.txt b/requirements_files/specific_golden_cpu.txt index 64c9ad9..dfb4e18 100644 --- a/requirements_files/specific_golden_cpu.txt +++ b/requirements_files/specific_golden_cpu.txt @@ -34,5 +34,7 @@ spacytextblob==0.1.7 pymilvus +keras_preprocessing + --extra-index-url https://download.pytorch.org/whl/cpu - torch \ No newline at end of file + torch diff --git a/requirements_files/specific_minimal_cpu.txt b/requirements_files/specific_minimal_cpu.txt index 267f2b1..394ac7d 100644 --- a/requirements_files/specific_minimal_cpu.txt +++ b/requirements_files/specific_minimal_cpu.txt @@ -9,3 +9,5 @@ mlflow==2.7.0 scipy scikit-learn networkx + +keras_preprocessing