Nick Canu commited on
Commit
25bc366
·
1 Parent(s): 231c112
Files changed (2) hide show
  1. description_generator.py +0 -2
  2. requirements.txt +2 -0
description_generator.py CHANGED
@@ -14,8 +14,6 @@ class input_manager:
14
  self.search_tokens = search_tokens
15
  self.key = dict(zip(list(key_df.columns),np.zeros(len(key_df.columns))))
16
  self.top_n = top_n
17
-
18
- spacy.cli.download("en_core_web_md")
19
  self.nlp = spacy.load("en_core_web_md")
20
 
21
  #translate input text to vector
 
14
  self.search_tokens = search_tokens
15
  self.key = dict(zip(list(key_df.columns),np.zeros(len(key_df.columns))))
16
  self.top_n = top_n
 
 
17
  self.nlp = spacy.load("en_core_web_md")
18
 
19
  #translate input text to vector
requirements.txt CHANGED
@@ -9,3 +9,5 @@ spacy==3.5.1
9
  streamlit==1.20.0
10
  torch==2.0.0
11
  transformers==4.27.3
 
 
 
9
  streamlit==1.20.0
10
  torch==2.0.0
11
  transformers==4.27.3
12
+ en_core_web_sm @ https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.4.0/en_core_web_sm-3.4.0-py3-none-any.whl
13
+ en_core_web_md @ https://github.com/explosion/spacy-models/releases/download/en_core_web_md-3.4.0/en_core_web_md-3.4.0-py3-none-any.whl