Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
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"]
|