Update pipeline.py

pipeline.py

@@ -1,20 +1,14 @@
+import json
 from utils_audio import convert_to_wav
 from to_cha import to_cha_from_wav
-from cha_json import cha_to_json_path
-from model_infer import AphasiaClassifier
-from output import format_result
+from cha_json import cha_to_json_file
+from output import predict_from_chajson
+
+MODEL_DIR = "./adaptive_aphasia_model"
 
 def run_pipeline(in_media_path: str, out_style: str = "json") -> str:
     wav = convert_to_wav(in_media_path, sr=16000, mono=True)
     cha = to_cha_from_wav(wav, lang="eng")
-    jsn = cha_to_json_path(cha)
-
-    clf = AphasiaClassifier()
-    pred = clf.predict_from_json(jsn)
-
-    return format_result(pred, style=out_style)
-
-if __name__ == "__main__":
-    import sys
-    media = sys.argv[1]
-    print(run_pipeline(media, out_style="text"))  # or "json"
+    chajson_path, _ = cha_to_json_file(cha)
+    results = predict_from_chajson(MODEL_DIR, chajson_path, output_file=None)
+    return json.dumps(results, ensure_ascii=False, indent=2)