Update requirements.txt

requirements.txt

@@ -1,12 +1,13 @@
 #------- Core dependencies -------
-torch>=2.0
+torch>=2.1.0
 transformers[sentencepiece]>=4.33,<4.34
 rich>=13.0.0,<14.0.0
-scikit-learn
-overrides
-pprintpp
-art
-colorama
+scikit-learn>=1.3,<1.4
+overrides>=7.4,<7.5
+art==6.1
+pprintpp==0.4.0
+colorama==0.4.6
+
 #------- Optional dependencies -------
 
 # train
@@ -16,6 +17,7 @@ hydra_colorlog
 wandb>=0.15,<0.16
 datasets>=2.13,<2.15
 
+
 # faiss
 faiss-cpu==1.7.4 # needed by: faiss
 
@@ -28,6 +30,17 @@ ipa-core # needed by: serve
 streamlit>=1.27,<1.28 # needed by: serve
 streamlit_extras>=0.3,<0.4 # needed by: serve
 
+fastapi>=0.104,<0.105 # needed by: serve, ray
+uvicorn[standard]==0.23.2 # needed by: serve, ray
+gunicorn==21.2.0 # needed by: serve, ray
+spacy>=3.7,<3.8 # needed by: serve, ray
+streamlit>=1.28,<1.29 # needed by: serve, ray
+streamlit_extras>=0.3,<0.4 # needed by: serve, ray
+pyvis # needed by: serve, ray
+ray[serve]>=2.7,<=2.8 # needed by: ray
+
+
+
 # retriever
 
 # reader