sync ms

app.py

@@ -17,6 +17,7 @@ from utils import (
     TEMP_DIR,
     TRANSLATE,
     CLASSES,
+    EN_US,
 )
 
 
@@ -94,7 +95,11 @@ def infer(wav_path: str, log_name: str, folder_path=TEMP_DIR):
         output: torch.Tensor = model(input)
         pred_id = torch.max(output.data, 1)[1]
         filename = os.path.basename(wav_path)
-        result = f"{TRANSLATE[CLASSES[pred_id]]} ({CLASSES[pred_id].capitalize()})"
+        result = (
+            CLASSES[pred_id].capitalize()
+            if EN_US
+            else f"{TRANSLATE[CLASSES[pred_id]]} ({CLASSES[pred_id].capitalize()})"
+        )
 
     except Exception as e:
         status = f"{e}"

utils.py

@@ -15,11 +15,6 @@ ZH2EN = {
     "中国五声调式识别": "Chinese pentatonic mode recognition",
     "建议录音时长保持在 20s 左右": "It is recommended to keep the recording length around 20s.",
     "引用": "Cite",
-    "宫": "Gong",
-    "商": "Shang",
-    "角": "Jue",
-    "徵": "Zhi",
-    "羽": "Yu",
 }
 
 MODEL_DIR = (
@@ -40,11 +35,11 @@ def _L(zh_txt: str):
 
 
 TRANSLATE = {
-    "Gong": _L("宫"),
-    "Shang": _L("商"),
-    "Jue": _L("角"),
-    "Zhi": _L("徵"),
-    "Yu": _L("羽"),
+    "Gong": "宫",
+    "Shang": "商",
+    "Jue": "角",
+    "Zhi": "徵",
+    "Yu": "羽",
 }
 CLASSES = list(TRANSLATE.keys())
 TEMP_DIR = "./__pycache__/tmp"