camparchimedes commited on
Commit
f9564f3
·
verified ·
1 Parent(s): d4b107b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -16,6 +16,7 @@ import numpy as np
16
  from fpdf import FPDF
17
  import time
18
  import os
 
19
 
20
  warnings.filterwarnings("ignore")
21
 
@@ -37,7 +38,7 @@ def transcribe_audio(audio_file):
37
  output = asr(audio_file, chunk_length_s=28, generate_kwargs={"num_beams": 5, "task": "transcribe", "language": "no"})
38
  return output["text"]
39
 
40
- # Create Gradio interface
41
  iface = gr.Interface(
42
  fn=transcribe_audio,
43
  inputs=gr.Audio(type="filepath"),
@@ -62,6 +63,8 @@ def convert_to_wav(audio_file):
62
  audio.export(wav_file, format="wav")
63
  return wav_file
64
 
 
 
65
  def transcribe_audio(audio_file, batch_size=4):
66
  start_time = time.time()
67
  # Convert .m4a to .wav
 
16
  from fpdf import FPDF
17
  import time
18
  import os
19
+ import spaces
20
 
21
  warnings.filterwarnings("ignore")
22
 
 
38
  output = asr(audio_file, chunk_length_s=28, generate_kwargs={"num_beams": 5, "task": "transcribe", "language": "no"})
39
  return output["text"]
40
 
41
+ # Gradio interface
42
  iface = gr.Interface(
43
  fn=transcribe_audio,
44
  inputs=gr.Audio(type="filepath"),
 
63
  audio.export(wav_file, format="wav")
64
  return wav_file
65
 
66
+
67
+ @spaces.GPU(duration=120, queue=False)
68
  def transcribe_audio(audio_file, batch_size=4):
69
  start_time = time.time()
70
  # Convert .m4a to .wav