abduaziz commited on
Commit
fdad199
1 Parent(s): 2d6c46c

Upload folder using huggingface_hub

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