Update DPTNet_eval/DPTNet_quant_sep.py

DPTNet_eval/DPTNet_quant_sep.py

@@ -36,17 +36,16 @@ def get_conf():
 def load_dpt_model():
     print('Load Separation Model...')
 
-    # 從環境變數取得 Hugging Face Token
-    HF_TOKEN = os.getenv("SpeechSeparation")
-    if not HF_TOKEN:
-        raise EnvironmentError("環境變數 HF_TOKEN 未設定！請先執行 export HF_TOKEN=xxx")
+# 從環境變數取得你設定的 Secret 名稱為 SpeechSeparation
+    speech_sep_token = os.getenv("SpeechSeparation")
+    if not speech_sep_token:
+        raise EnvironmentError("環境變數 SpeechSeparation 未設定！")
 
-    # 從 Hugging Face Hub 下載模型權重
     model_path = hf_hub_download(
-        repo_id="DeepLearning101/speech-separation",  # ← 替換成你的 repo 名稱
-        filename="train_dptnet_aishell_partOverlap_B6_300epoch_quan-int8.p",
-        token=SpeechSeparation
-    )
+    repo_id="DeepLearning101/speech-separation",
+    filename="train_dptnet_aishell_partOverlap_B2_300epoch_quan-int8.p",
+    token=speech_sep_token  # ✅ 使用你設定的 Secret
+    )    
 
     # 取得模型參數
     conf_filterbank, conf_masknet = get_conf()