Pranjal12345 commited on
Commit
0307cc0
1 Parent(s): ca784e9
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -226,7 +226,7 @@ model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-small")
226
  model.config.forced_decoder_ids = None
227
 
228
  chunk_duration = 30
229
- overlap_duration = 5
230
 
231
  @app.get("/")
232
  def read_root():
@@ -268,5 +268,5 @@ async def transcribe_audio(audio_file: UploadFile):
268
  final_transcription = remove_tags(final_transcription)
269
 
270
 
271
- return {"transcription": final_transcription}
272
 
 
226
  model.config.forced_decoder_ids = None
227
 
228
  chunk_duration = 30
229
+ overlap_duration = 3
230
 
231
  @app.get("/")
232
  def read_root():
 
268
  final_transcription = remove_tags(final_transcription)
269
 
270
 
271
+ return final_transcription
272