update

app.py

@@ -55,11 +55,11 @@ from process_1 import SynthradAlgorithm1
 # Streamlit UI
 # =========================
 st.set_page_config(page_title="SynthRad (nnUNetv2) Demo", layout="wide")
-st.title("SynthRad — 双任务演示（MRI/CBCT + Mask → synthetic CT）")
+st.title("SynthRad — MRI/CBCT + Mask → synthetic CT")
 
 # 任务选择
 TASKS = ["Task 1 (MR → CT)", "Task 2 (CBCT → CT)"]
-task = st.radio("选择任务", TASKS, index=0, horizontal=True)
+task = st.radio("Select Task", TASKS, index=0, horizontal=True)
 
 # 根据任务设置标题/提示
 if task == "Task 1 (MR → CT)":
@@ -111,7 +111,7 @@ def build_sample_map(task_name: str):
     repo_dir = REPO_DIRS[task_name]
     if task_name == "Task 1 (MR → CT)":
         vol_key = "mri"
-        vol_fname = "mri.mha"    # 如果你的样例文件名不是 mri.mha，请改成实际名称
+        vol_fname = "mr.mha"    # 如果你的样例文件名不是 mri.mha，请改成实际名称
         mask_fname = "mask1.mha"    # 如果你的样例文件名不是 mri.mha，请改成实际名称
     else:
         vol_key = "cbct"
@@ -261,7 +261,7 @@ else:
     # 下载输入体积（根据任务区分命名）
     with col_d1:
         if st.session_state.input_vol is not None:
-            in_name = "input_mri.nii.gz" if task == "Task 1 (MR → CT)" else "input_cbct.nii.gz"
+            in_name = "input_mr.nii.gz" if task == "Task 1 (MR → CT)" else "input_cbct.nii.gz"
             in_label = "Download input MRI" if task == "Task 1 (MR → CT)" else "Download input CBCT"
             _download_sitk_image(
                 st.session_state.input_vol,

app_1.py

@@ -3,7 +3,7 @@ import os
 from huggingface_hub import snapshot_download
 
 from PIL import Image, ImageDraw
-HF_REPO = "Synthard2025KoalAI/synthrad2025_task1"   
+HF_REPO = "aehrc/Synthrad2025"   
 LOCAL_WEIGHTS_DIR = os.path.abspath("weights/task1")  
 
 token = os.getenv("HF_TOKEN")