Toumaima commited on
Commit
9fcccf9
·
verified ·
1 Parent(s): abab189

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -16,16 +16,16 @@ from spacy.cli import download
16
  class BasicAgent:
17
  def __init__(self):
18
  print("BasicAgent initialized.")
 
 
 
 
 
19
  self.whisper_model = whisper.load_model("base")
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
- download("en_core_web_sm")
27
- self.spacy = spacy.load("en_core_web_sm")
28
-
29
  def extract_named_entities(self, text):
30
  entities = self.ner_pipeline(text)
31
  return [e["word"] for e in entities if e["entity_group"] == "PER"]
 
16
  class BasicAgent:
17
  def __init__(self):
18
  print("BasicAgent initialized.")
19
+ try:
20
+ self.spacy = spacy.load("en_core_web_sm")
21
+ except OSError:
22
+ download("en_core_web_sm")
23
+ self.spacy = spacy.load("en_core_web_sm")
24
  self.whisper_model = whisper.load_model("base")
25
  self.qa_pipeline = pipeline("question-answering")
26
  self.ner_pipeline = pipeline("ner", aggregation_strategy="simple")
27
  self.embedding_model = pipeline("feature-extraction")
28
+
 
 
 
 
 
29
  def extract_named_entities(self, text):
30
  entities = self.ner_pipeline(text)
31
  return [e["word"] for e in entities if e["entity_group"] == "PER"]