oza75 commited on
Commit
4c75eaa
·
verified ·
1 Parent(s): f49dcf2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -84,16 +84,15 @@ def transcribe(audio, task_type, language):
84
 
85
  # Use the pipeline to perform transcription
86
  sample = {"array": waveform.squeeze().numpy(), "sampling_rate": sample_rate}
87
- texts = pipe(sample, generate_kwargs={
88
  "task": task_type,
89
  "num_beams": 3,
90
- "num_return_sequences": 3,
91
  "no_repeat_ngram_size": 4,
92
  "early_stopping": True,
93
  "language": language
94
- })
95
- print(texts)
96
- return texts[0]['text']
97
 
98
  def get_wav_files(directory):
99
  """
 
84
 
85
  # Use the pipeline to perform transcription
86
  sample = {"array": waveform.squeeze().numpy(), "sampling_rate": sample_rate}
87
+ text = pipe(sample, generate_kwargs={
88
  "task": task_type,
89
  "num_beams": 3,
 
90
  "no_repeat_ngram_size": 4,
91
  "early_stopping": True,
92
  "language": language
93
+ })["text"]
94
+
95
+ return text
96
 
97
  def get_wav_files(directory):
98
  """