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):
|