适配zeroGPU

app.py

@@ -58,9 +58,11 @@ def init_model_my():
     logging.basicConfig(level=logging.DEBUG,
                         format='%(asctime)s %(levelname)s %(message)s')
     config_path = "train.yaml"
-    checkpoint_path = "step_32499.pt"
+    from huggingface_hub import hf_hub_download
+    # 从Hugging Face下载.pt文件
+    pt_file_path = hf_hub_download(repo_id="ASLP-lab/OSUM", filename="infer.pt")
     args = SimpleNamespace(**{
-        "checkpoint": checkpoint_path,
+        "checkpoint": pt_file_path,
     })
     configs = load_dict_from_yaml(config_path)
     model, configs = init_model(args, configs)
@@ -69,7 +71,7 @@ def init_model_my():
     print(model)
     return model, tokenizer
 
-# model, tokenizer = init_model_my()
+model, tokenizer = init_model_my()
 print("model init success")
 def do_resample(input_wav_path, output_wav_path):
     """"""