diff --git a/ieemoo-ai-predict.py b/ieemoo-ai-predict.py
index 7b4e611..534bf45 100644
--- a/ieemoo-ai-predict.py
+++ b/ieemoo-ai-predict.py
@@ -37,7 +37,7 @@ app = Flask(__name__)
 load model
 '''
 parser = argparse.ArgumentParser()
-parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-predict/', help='model.pt path(s)')
+parser.add_argument('--weights', nargs='+', type=str, default='../module/ieemoo-ai-predict/model/now/', help='model.pt path(s)')
 parser.add_argument('--Form', type=str, default='')
 parser.add_argument('--source', type=str, default='../module/ieemoo-ai-predict/imgs', help='source')  # file/folder, 0 for webcam
 parser.add_argument('--img-size', type=int, default=640, help='inference size (pixels)')
diff --git a/requirements.txt b/requirements.txt
index 963530c..b3b7c3f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,10 +5,10 @@ matplotlib==3.4.1
 numpy==1.20.2
 esdk-obs-python --trusted-host pypi.org
 opencv_python==4.5.1.48
+opencv-contrib-python==4.5.5.64
 Pillow==9.1.0
 scipy==1.6.2
 setuptools==49.6.0
-ml-collections==0.1.1
 coremltools==5.2.0
 onnx==1.7.0
 pandas==1.2.4
@@ -16,4 +16,6 @@ pycocotools==2.0.2
 PyYAML==6.0
 requests==2.25.1
 seaborn==0.11.1
-tqdm==4.60.0
\ No newline at end of file
+thop==0.0.31.post2005241907
+tqdm==4.60.0
+ml-collections==0.1.1