[Update] model_file check json first

10 months ago · 2d9650662f
parent 29ec0a68b2
commit 2d9650662f
1 changed files with 3 additions and 4 deletions
--- a/paddlespeech/cls/exps/panns/deploy/predict.py
+++ b/paddlespeech/cls/exps/panns/deploy/predict.py
@ -74,11 +74,10 @@ class Predictor(object):
                 enable_mkldnn=False):
        self.batch_size = batch_size

-        model_file = os.path.join(model_dir, "inference.pdmodel")
-        if not os.path.exists(model_file):
+        if os.path.exists(os.path.join(model_dir, "inference.json")):
            model_file = os.path.join(model_dir, "inference.json")
-            if not os.path.exists(model_file):
-                raise ValueError("Inference model file not exists!")
+        else:
+            model_file = os.path.join(model_dir, "inference.pdmodel")

        params_file = os.path.join(model_dir, "inference.pdiparams")