Toumaima commited on
Commit
451bf10
·
verified ·
1 Parent(s): aea3558

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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):