Save Operation for CLI

cabelo · May 8, 2022 · 2d0aad0 · 2d0aad0
1 parent a7cbf6d
commit 2d0aad0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py
@@ -824,4 +824,6 @@ def detectFace(img_path, target_size = (224, 224), detector_backend = 'opencv',
 #main
 
 functions.initialize_folder()
-fire.Fire()
+
+def cli():
+	fire.Fire()
diff --git a/setup.py b/setup.py
@@ -21,7 +21,7 @@
     ],
     entry_points={
         "console_scripts":
-        ["DeepFace = deepface:DeepFace"],
+        ["deepface = deepface:DeepFace.cli"],
     },    
     python_requires='>=3.5.5',
     install_requires=["numpy>=1.14.0", "pandas>=0.23.4", "tqdm>=4.30.0", "gdown>=3.10.1", "Pillow>=5.2.0", "opencv-python>=3.4.4", "tensorflow>=1.9.0", "keras>=2.2.0", "Flask>=1.1.2", "mtcnn>=0.1.0", "retina-face>=0.0.1", "fire>=0.4.0"]