Michael Natanael commited on
Commit
5d3a20e
·
1 Parent(s): 4a4c6b2

add chunk_length_s for longer audio

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -69,6 +69,8 @@ def whisper_api(temp_audio_path):
69
  model=model,
70
  tokenizer=processor.tokenizer,
71
  feature_extractor=processor.feature_extractor,
 
 
72
  torch_dtype=torch_dtype,
73
  device=device,
74
  )
 
69
  model=model,
70
  tokenizer=processor.tokenizer,
71
  feature_extractor=processor.feature_extractor,
72
+ chunk_length_s=30,
73
+ batch_size=32, # batch size for inference - set based on your device
74
  torch_dtype=torch_dtype,
75
  device=device,
76
  )