requirements.txt

--index-url https://thestage.jfrog.io/artifactory/api/pypi/pypi-thestage-ai-production/simple
--extra-index-url https://pypi.nvidia.com
--extra-index-url https://pypi.org/simple

torch
thestage
elastic_models[nvidia]
scipy
transformers