environment.yml

name: QADemo
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - cudatoolkit=11.6.0
  - numpy-base=1.23.1
  - pip=22.2.2
  - python=3.10.6
  - pytorch=1.12.1
  - torchaudio=0.12.1
  - torchvision=0.13.1
  - pip:
    - en-core-web-sm @ https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.2.0/en_core_web_sm-3.2.0-py3-none-any.whl
    - huggingface-hub==0.10.0
    - nltk==3.7
    - numpy==1.23.3
    - pandas==1.5.0
    - scikit-learn==1.1.2
    - scipy==1.9.2
    - sentence-transformers==2.2.2
    - spacy==3.2.0
    - sentencepiece==0.1.97
    - streamlit==1.13.0
    - streamlit-aggrid==0.3.3
    - tokenizers==0.12.1
    - toml==0.10.2
    - toolz==0.12.0
    - tqdm==4.64.1
    - transformers==4.22.2