diff --git a/yolov5.py b/yolov5.py
index fed9d58..5a0ca75 100644
--- a/yolov5.py
+++ b/yolov5.py
@@ -1,5 +1,6 @@
 import torch
 import numpy
+from pathlib import Path
 
 from towhee import register
 from towhee.operator import NNOperator