Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,12 @@ import docx
|
|
9 |
import json
|
10 |
|
11 |
# Load models
|
12 |
-
|
|
|
|
|
|
|
|
|
|
|
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 |
|