Better import

Author: lolipopshock

src/layoutparser/__init__.py

@@ -12,4 +12,8 @@
 from .ocr import (
     GCVFeatureType, GCVAgent, 
     TesseractFeatureType, TesseractAgent
+)
+
+from .models import (
+    Detectron2LayoutModel
 )

src/layoutparser/models.py

@@ -5,6 +5,8 @@
 from detectron2.engine import DefaultPredictor
 from .elements import *
 
+__all__ = ['Detectron2LayoutModel']
+
 
 class BaseLayoutModel(ABC):