refactor: remove unused fix_speaker function and clean up code in onnx_infer.py

3 files changed, 4 insertions, 12 deletions

diff --git a/models/trpg-final/model.onnx b/models/trpg-final/model.onnx
index e5e585c..b125d02 100644
--- a/models/trpg-final/model.onnx
+++ b/models/trpg-final/model.onnx
diff --git a/models/trpg-final/model.onnx.data b/models/trpg-final/model.onnx.data
index 9465ee4..fa5b018 100644
--- a/models/trpg-final/model.onnx.data
+++ b/models/trpg-final/model.onnx.data
diff --git a/tests/onnx_infer.py b/tests/onnx_infer.py
index d6c00d2..31cc546 100644
--- a/tests/onnx_infer.py
+++ b/tests/onnx_infer.py
@@ -116,14 +116,6 @@ def fix_timestamp(ts):
     return ts
 
 
-def fix_speaker(spk):
-    if not spk:
-        return spk
-    spk = re.sub(r"[^\w\s\u4e00-\u9fff]+$", "", spk)
-    if len(spk) == 1 and re.match(r"^[风雷电雨雪火水木金]", spk):
-        return spk + "某"
-    return spk
-
 
 out = {"metadata": {}, "content": []}
 for e in entities:
@@ -132,10 +124,10 @@ for e in entities:
     conf = round(float(np.mean(e["probs"])), 3)
     typ = e["type"]
     if typ in ("timestamp", "speaker"):
-        if typ == "timestamp":
-            ent_text = fix_timestamp(ent_text)
-        else:
-            ent_text = fix_speaker(ent_text)
+        # if typ == "timestamp":
+        #     ent_text = fix_timestamp(ent_text)
+        # else:
+        #     ent_text = ent_text
         out["metadata"][typ] = ent_text
     else:
         out["content"].append({"type": typ, "content": ent_text, "confidence": conf})