Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -20,6 +20,11 @@ class BasicAgent:
|
|
20 |
self.qa_pipeline = pipeline("question-answering")
|
21 |
self.ner_pipeline = pipeline("ner", aggregation_strategy="simple")
|
22 |
self.embedding_model = pipeline("feature-extraction")
|
|
|
|
|
|
|
|
|
|
|
23 |
self.spacy = spacy.load("en_core_web_sm")
|
24 |
|
25 |
def extract_named_entities(self, text):
|
|
|
20 |
self.qa_pipeline = pipeline("question-answering")
|
21 |
self.ner_pipeline = pipeline("ner", aggregation_strategy="simple")
|
22 |
self.embedding_model = pipeline("feature-extraction")
|
23 |
+
try:
|
24 |
+
self.spacy = spacy.load("en_core_web_sm")
|
25 |
+
except OSError:
|
26 |
+
from spacy.cli import download
|
27 |
+
download("en_core_web_sm")
|
28 |
self.spacy = spacy.load("en_core_web_sm")
|
29 |
|
30 |
def extract_named_entities(self, text):
|