diff --git a/deploy/python/predict_cls.py b/deploy/python/predict_cls.py
index e4ff254150113fa005208ac89d99c939043e6ffd..6c312b423345967a1f163f09d79ab82f4ea83c72 100644
--- a/deploy/python/predict_cls.py
+++ b/deploy/python/predict_cls.py
@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import os
+
 import cv2
 import numpy as np