Update requirements.txt

requirements.txt

@@ -1,107 +1,13 @@
-argon2-cffi==20.1.0
-async-generator==1.10
-attrs==20.3.0
-backcall==0.2.0
-bleach==3.3.0
-certifi==2020.12.5
-cffi==1.14.5
-chardet==3.0.4
-click==7.1.2
-cycler==0.10.0
-Cython==0.29.22
-dataclasses==0.6
-decorator==4.4.2
-defusedxml==0.6.0
-demjson==2.2.4
-editdistance==0.5.3
-einops==0.3.0
-entrypoints==0.3
-fasttext==0.9.1
-filelock==3.0.12
-future==0.18.2
-gitdb==4.0.5
-GitPython==3.1.0
-idna==2.10
-imageio==2.9.0
-importlib-metadata==3.4.0
-ipykernel==5.4.3
-ipython==7.20.0
-ipython-genutils==0.2.0
-ipywidgets==7.6.3
-jedi==0.18.0
-Jinja2==2.11.3
-joblib==0.17.0
-jsonschema==3.2.0
-jupyter-client==6.1.11
-jupyter-console==6.2.0
-jupyter-core==4.7.1
-jupyterlab-pygments==0.1.2
-jupyterlab-widgets==1.0.0
-kiwisolver==1.3.1
-lmdb==0.98
-MarkupSafe==1.1.1
-matplotlib==3.3.4
-mistune==0.8.4
-nbclient==0.5.2
-nbconvert==6.0.7
-nbformat==5.1.2
-nest-asyncio==1.5.1
-networkx==2.4
-nltk==3.4.5
-notebook==6.2.0
-numpy==1.19.2
-omegaconf==2.0.1rc4
-opencv-python==4.5.1.48
-packaging==20.9
-pandocfilters==1.4.3
-parso==0.8.1
-pexpect==4.8.0
-pickleshare==0.7.5
-Pillow==8.1.2
-prometheus-client==0.9.0
-prompt-toolkit==3.0.16
-protobuf==3.15.6
-ptyprocess==0.7.0
-pybind11==2.6.2
-pycocotools==2.0.2
-pycparser==2.20
-pyparsing==2.4.7
-pyrsistent==0.17.3
-python-dateutil==2.8.1
-PyWavelets==1.1.1
-PyYAML==5.4.1
-pyzmq==22.0.3
-qtconsole==5.0.2
-QtPy==1.9.0
-regex==2020.11.13
-requests==2.23.0
-sacremoses==0.0.43
-scikit-image==0.17.2
-scikit-learn==0.23.2
-scipy==1.6.1
-Send2Trash==1.5.0
-sentencepiece==0.1.91
-six==1.15.0
-sklearn==0.0
-smmap==3.0.5
-termcolor==1.1.0
-terminado==0.9.2
-testpath==0.4.4
-threadpoolctl==2.1.0
-tifffile==2021.2.1
-tokenizers==0.9.3
-torch==1.7.1
-torchtext==0.5.0
-torchvision==0.8.2
-tornado==6.1
-tqdm==4.51.0
-traitlets==5.0.5
+einops
+numpy
+matplotlib
+opencv-python
+h5py
+torch
+torchvision
 transformers==3.5.1
-typing-extensions==3.7.4.3
-urllib3==1.25.11
-utils==1.0.1
-wcwidth==0.2.5
-webencodings==0.5.1
-wget==3.2
-widgetsnbextension==3.5.1
-zipp==3.4.0
+gensim
+tensorflow
+captum
+ipython
+wget