Spaces:
Runtime error
Runtime error
Nick Canu
commited on
Commit
·
25bc366
1
Parent(s):
231c112
spacy fix
Browse files- description_generator.py +0 -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
|