Merge pull request #2 from ConvLab/update_env

Update package requirement in setup.py

setup.py

@@ -31,37 +31,26 @@ setup(
                 'Topic :: Scientific/Engineering :: Artificial Intelligence',
     ],
     install_requires=[
-        'nltk>=3.4',
-        'tqdm>=4.30',
-        'checksumdir>=1.1',
-        'dataclasses',
-        'visdom',
-        'Pillow',
-        'future',
-        'torch>=1.2.0,<1.6.0',
-        'numpy>=1.15.0',
-        'scipy',
-        'scikit-learn==0.20.3',
-        'pytorch-pretrained-bert>=0.6.1',
-        'transformers>=2.3.0,<3.0.0',
-        'tensorboard>=1.14.0',
-        'tensorboardX==1.7',
-        'tokenizers>=0.8.0',
-        'overrides==4.1.2',
-        'allennlp==0.9.0',
+        'tabulate',
+        'python-Levenshtein',
         'requests',
+        'numpy',
+        'nltk',
+        'scipy',
+        'torch>=1.10.0',
+        'transformers>=4.0',
+        'spacy',
+        'allennlp',
         'simplejson',
-        'spacy==2.1.9',
         'unidecode',
         'jieba',
         'embeddings',
+        'visdom',
         'quadprog',
-        'pyyaml',
         'fuzzywuzzy',
-        'python-Levenshtein',
         'json_lines',
         'gtts',
-        'DeepSpeech',
+        'deepspeech',
         'pydub'
     ],
     extras_require={