From d2c1e5e10fb7660f745a154265f824b014b7d6b2 Mon Sep 17 00:00:00 2001
From: TonyLee1256 <163754792+TonyLee1256@users.noreply.github.com>
Date: Tue, 7 May 2024 13:28:07 +0800
Subject: [PATCH] bugfix inference.py

---
 src/inference.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/inference.py b/src/inference.py
index 4022f6c..314a934 100644
--- a/src/inference.py
+++ b/src/inference.py
@@ -55,6 +55,9 @@ if __name__ == '__main__':
     if args.mix and args.lang == "None":
         print("When -mix is set, -lang must be set (support: ['zh', 'en'])")
         sys.exit(-1)
+    elif args.mix and args.lang not in ['zh', 'en']:
+        print(f"language support: ['zh', 'en'] (invalid: {args.lang})")
+        sys.exit(-1)
     
     # You can use your own checkpoint and tokenizer path.
     print('Loading model and tokenizer...')