Update app.py

app.py

@@ -5,18 +5,16 @@ REPO_TYPE = "hf"
 if REPO_TYPE not in ["hf", "ms"]:
     raise ValueError("REPO_TYPE must be either 'hf' for Hugging Face or 'ms' for ModelScope.")
 
-
-from huggingface_hub import snapshot_download
+if REPO_TYPE == "hf":
+    from huggingface_hub import snapshot_download
+else:
+    from modelscope.hub.snapshot_download import snapshot_download
 
 
 
 # 1. 定义本地路径和远程仓库ID
 FUN_ASR_NANO_LOCAL_PATH = "./Fun-ASR/model"
-FUN_ASR_NANO_REPO_ID = "FunAudioLLM/Fun-ASR-Nano-2512"
 SENSE_VOICE_SMALL_LOCAL_PATH = "./Fun-ASR/model/SenseVoiceSmall"
-# SENSE_VOICE_SMALL_REPO_ID = "FunAudioLLM/SenseVoiceSmall"
-# REPO_TYPE = "hf"  # "hf" for Hugging Face, "ms" for ModelScope
-SENSE_VOICE_SMALL_REPO_ID = "iic/SenseVoiceSmall"
 
 
 if REPO_TYPE == "ms":
@@ -408,7 +406,7 @@ def get_model_options(pipeline_type):
 # Dictionary to store loaded models
 loaded_models = {}
 
-@spaces.GPU(duration=120)
+@spaces.GPU()
 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):
     """
     Transcribes audio from a given source using SenseVoice.