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"]
|