martyn-wong commited on
Commit
654e8ae
·
1 Parent(s): d590189

updated utils.py

Browse files
Files changed (1) hide show
  1. util.py +7 -1
util.py CHANGED
@@ -14,10 +14,16 @@ from dotenv import load_dotenv
14
  from openai import OpenAI
15
  from typing import List
16
 
 
17
  sch = SemanticScholar(timeout=30)
18
  cr = Crossref(mailto="[email protected]")
19
 
20
- nlp = spacy.load("en_core_web_sm")
 
 
 
 
 
21
  kw_model = KeyBERT(model="sentence-transformers/allenai-specter")
22
  embed_model = SentenceTransformer("sentence-transformers/allenai-specter")
23
 
 
14
  from openai import OpenAI
15
  from typing import List
16
 
17
+
18
  sch = SemanticScholar(timeout=30)
19
  cr = Crossref(mailto="[email protected]")
20
 
21
+ try:
22
+ nlp = spacy.load("en_core_web_sm")
23
+ except OSError:
24
+ spacy.cli.download("en_core_web_sm")
25
+ nlp = spacy.load("en_core_web_sm")
26
+
27
  kw_model = KeyBERT(model="sentence-transformers/allenai-specter")
28
  embed_model = SentenceTransformer("sentence-transformers/allenai-specter")
29