diff --git a/base/requirements.txt b/base/requirements.txt index f5b1100..cf703e6 100644 --- a/base/requirements.txt +++ b/base/requirements.txt @@ -2,6 +2,6 @@ bottle==0.12.25 bottle-log==1.0.0 certifi==2023.7.22 charset-normalizer==3.3.0 -idna==3.4 -requests==2.31.0 -urllib3==2.2.1 +idna==3.7 +requests==2.32.2 +urllib3==2.2.2 diff --git a/pie/base/README.md b/pie/base/README.md new file mode 100644 index 0000000..27bbab0 --- /dev/null +++ b/pie/base/README.md @@ -0,0 +1,3 @@ +The `./requirements.txt` in this folder are the needed requirements to tag a file with pie. + +The `./pie/requirements.txt` in the `pie/` folder below are needed to train pie. This needs to install a lot more pytorch & cuda stuff (because unlike just tagging a file, we need a GPU to train). We don't want this to pollute the docker container, so instead we use the lighter `./requirements.txt`. \ No newline at end of file diff --git a/pie/base/requirements.txt b/pie/base/requirements.txt index d397888..f136ae6 100644 --- a/pie/base/requirements.txt +++ b/pie/base/requirements.txt @@ -3,7 +3,7 @@ click==8.1.7 filelock==3.13.1 fsspec==2023.12.1 gensim==4.3.2 -Jinja2==3.1.3 +Jinja2==3.1.4 joblib==1.3.2 JSON-minify==0.3.0 lxml==4.9.3 @@ -12,7 +12,7 @@ mpmath==1.3.0 networkx==3.2.1 numpy==1.26.2 PyYAML==6.0.1 -scikit-learn==1.3.2 +scikit-learn==1.5.0 scipy==1.11.4 smart-open==6.4.0 sympy==1.12 @@ -20,6 +20,6 @@ termcolor==2.4.0 terminaltables==3.1.10 threadpoolctl==3.2.0 torch==2.1.1+cpu -tqdm==4.66.1 +tqdm==4.66.3 typing==3.7.4.3 typing_extensions==4.8.0