oza75 commited on
Commit
f49dcf2
·
verified ·
1 Parent(s): 8d4eba5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -84,9 +84,16 @@ 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
- text = pipe(sample, generate_kwargs={"task": task_type, "num_beams": 10, "early_stopping": True, "language": language})["text"]
88
-
89
- return text
 
 
 
 
 
 
 
90
 
91
  def get_wav_files(directory):
92
  """
 
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
  """