Akchhaya commited on
Commit
a6890b8
·
verified ·
1 Parent(s): 1a83f20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -9,7 +9,12 @@ import docx
9
  import json
10
 
11
  # Load models
12
- nlp = spacy.load("en_core_web_sm")
 
 
 
 
 
13
  keyword_extractor = pipeline("token-classification", model="jean-baptiste/roberta-large-ner-english")
14
  classifier = pipeline("text-classification", model="microsoft/MiniLM-L12-H384-uncased")
15
 
 
9
  import json
10
 
11
  # Load models
12
+ try:
13
+ nlp = spacy.load("en_core_web_sm")
14
+ except OSError:
15
+ from spacy.cli import download
16
+ download("en_core_web_sm")
17
+ nlp = spacy.load("en_core_web_sm")
18
  keyword_extractor = pipeline("token-classification", model="jean-baptiste/roberta-large-ner-english")
19
  classifier = pipeline("text-classification", model="microsoft/MiniLM-L12-H384-uncased")
20