diff --git a/__init__.py b/__init__.py
index f459787..a635800 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,10 +1,3 @@
-import os
-try:
-    import detectron2
-except ModuleNotFoundError:
-    os.system("git clone https://github.com/facebookresearch/detectron2.git")
-    os.system("python -m pip install -e detectron2")
-
 from .detectron import Detectron2
 
 
diff --git a/detectron.py b/detectron.py
index c3539ce..5e36607 100644
--- a/detectron.py
+++ b/detectron.py
@@ -1,5 +1,10 @@
 from typing import List, Tuple
-
+import os
+try:
+    import detectron2
+except ModuleNotFoundError:
+    os.system("git clone https://github.com/facebookresearch/detectron2.git")
+    os.system("python -m pip install -e detectron2")
 from detectron2 import model_zoo
 from detectron2.config import get_cfg
 from detectron2.engine.defaults import DefaultPredictor