Ngoufack commited on
Commit
3270fd1
·
1 Parent(s): ef46332
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def transcribe(inputs, task):
19
  raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")
20
 
21
  audio = whisperx.load_audio(inputs)
22
- result = model.transcribe(audio, batch_size=BATCH_SIZE)
23
  model_a, metadata = whisperx.load_align_model(language_code=result["language"], device=device)
24
  result = whisperx.align(result["segments"], model_a, metadata, audio, device, return_char_alignments=False)
25
  diarize_model = whisperx.DiarizationPipeline(use_auth_token=os.getenv("HF_TOKEN"), device=device)
 
19
  raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")
20
 
21
  audio = whisperx.load_audio(inputs)
22
+ result = model.transcribe(audio, batch_size=BATCH_SIZE,compute_type="int8")
23
  model_a, metadata = whisperx.load_align_model(language_code=result["language"], device=device)
24
  result = whisperx.align(result["segments"], model_a, metadata, audio, device, return_char_alignments=False)
25
  diarize_model = whisperx.DiarizationPipeline(use_auth_token=os.getenv("HF_TOKEN"), device=device)