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