FFomy commited on
Commit
42143e2
·
verified ·
1 Parent(s): 2a9b280

add 'import spaces' and add

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,4 +1,5 @@
1
  import os
 
2
  # from huggingface_hub import snapshot_download
3
  from modelscope.hub.snapshot_download import snapshot_download
4
 
@@ -401,6 +402,8 @@ def get_model_options(pipeline_type):
401
  # Dictionary to store loaded models
402
  loaded_models = {}
403
 
 
 
404
  def transcribe_audio(audio_input, audio_url, proxy_url, proxy_username, proxy_password, pipeline_type, model_id, download_method, start_time=None, end_time=None, verbose=False):
405
  """
406
  Transcribes audio from a given source using SenseVoice.
 
1
  import os
2
+ import spaces
3
  # from huggingface_hub import snapshot_download
4
  from modelscope.hub.snapshot_download import snapshot_download
5
 
 
402
  # Dictionary to store loaded models
403
  loaded_models = {}
404
 
405
+
406
+ @spaces.GPU
407
  def transcribe_audio(audio_input, audio_url, proxy_url, proxy_username, proxy_password, pipeline_type, model_id, download_method, start_time=None, end_time=None, verbose=False):
408
  """
409
  Transcribes audio from a given source using SenseVoice.