Upload folder using huggingface_hub
Browse files
pipe.py
CHANGED
@@ -87,11 +87,7 @@ class AudioSpeechNERPipeline:
|
|
87 |
# Extract named entities
|
88 |
entities = self.ner_pipeline(transcription)
|
89 |
|
90 |
-
return
|
91 |
-
'filename': os.path.basename(audio_path),
|
92 |
-
'transcription': transcription,
|
93 |
-
'entities': entities
|
94 |
-
}
|
95 |
|
96 |
def create_ner_html(entities):
|
97 |
"""
|
|
|
87 |
# Extract named entities
|
88 |
entities = self.ner_pipeline(transcription)
|
89 |
|
90 |
+
return transcription, entities
|
|
|
|
|
|
|
|
|
91 |
|
92 |
def create_ner_html(entities):
|
93 |
"""
|