diff --git a/apps/ai/main.py b/apps/ai/main.py
index 36831117..f7512b46 100644
--- a/apps/ai/main.py
+++ b/apps/ai/main.py
@@ -6,11 +6,11 @@ myLogger = logging.getLogger('log')
 
 
 algo_dict = {
-    "helmet": "apps.helmet.algo.helmet"
+    "helmet": "apps.ai.helmet.algo.helmet"
 }
 
 
-def ai_analyse(codes: list, global_img: str, face_img: str=None):
+def ai_analyse(codes: list, global_img: str, face_img: str = None):
     """多线程进行算法分析图片
 
     Args: