JacobLinCool commited on
Commit
5f56a57
·
1 Parent(s): 2a3f2ed

feat: add GPU decorator to transcribe_audio_phi function

Browse files
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -146,6 +146,7 @@ def load_phi_model() -> None:
146
  raise
147
 
148
 
 
149
  def transcribe_audio_phi(audio: str) -> str:
150
  """
151
  Transcribe audio using the Phi-4 model.
 
146
  raise
147
 
148
 
149
+ @spaces.GPU(duration=get_gpu_duration)
150
  def transcribe_audio_phi(audio: str) -> str:
151
  """
152
  Transcribe audio using the Phi-4 model.