Update main console

src/deep_package_detection/main.py

@@ -64,7 +64,6 @@ def train(
     )
     detector.train()
     _ = detector.validation()
-    detector.model_export()
 
 
 @deep_package_detection_cli.command()
@@ -85,7 +84,7 @@ def train(
 )
 def infer(model_path: Path, data_path: str, result_path: str) -> None:
     """This the CLI for testing purposes"""
-    logger.info("Testing the YOLO model for package detection...")
+    logger.info("Testing the model for package detection...")
     inferer = PackageDetectorInference(
         model_path=Path(model_path), result_path=result_path
     )