diff --git a/faced/detector.py b/faced/detector.py
index 330fbaf..fec85db 100644
--- a/faced/detector.py
+++ b/faced/detector.py
@@ -1,4 +1,7 @@
-import tensorflow as tf
+try:
+    import tensorflow.compat.v1 as tf
+except ImportError:
+    import tensorflow as tf
 import cv2
 import numpy as np
 import os