Rename entrypoint.py to main.py

entrypoint.py → main.py

@@ -1,17 +1,13 @@
-from __future__ import annotations
-
 import argparse
 import json
 import sys
 from io import BytesIO
 from pathlib import Path
-from typing import Any, Dict, List, TYPE_CHECKING
+from typing import Any, Dict, List
 
 from PIL import Image
 from ultralytics import YOLO
-
-if TYPE_CHECKING:
-    import numpy as np
+import numpy as np
 
 
 def load_image(frame: Any, base_dir: Path) -> Image.Image:
@@ -70,7 +66,7 @@ def build_parser() -> argparse.ArgumentParser:
     return parser
 
 
-def main() -> None:
+if __name__ == "__main__":
     args = build_parser().parse_args()
 
     base_dir = Path(__file__).resolve().parent
@@ -85,12 +81,6 @@ def main() -> None:
         print("[]")
         return
 
-    import numpy as np
-
     frame = np.array(image)
     output = run_model(model, frame)
-    print(json.dumps(output))
-
-
-if __name__ == "__main__":
-    main()
+    print(json.dumps(output))