diff --git a/main.py b/main.py
index fa58a09..290202b 100644
--- a/main.py
+++ b/main.py
@@ -128,7 +128,7 @@
     ears_ai = EarsAI()
     calc_position = CalcStethoscopePosition()
     base_dir = os.path.join(args.output_dir, "frames")
-    results_dir = os.path.join(os.path.dirname(args.output_dir), "results")
+    results_dir = os.path.join(args.output_dir, "results")
     csv_path = os.path.join(results_dir, "results.csv")
     normalized_csv_path = os.path.join(results_dir, "results-convert.csv")
     pose_overlay_dir = os.path.join(results_dir, "pose_overlay_image")
diff --git a/requirements.txt b/requirements.txt
index 6437e1d..71488a4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,4 +6,5 @@
 lightgbm == 4.5.0
 xgboost == 2.1.1
 scipy == 1.9.3
-numpy == 1.24.0
\ No newline at end of file
+numpy == 1.24.0
+scikit-learn == 1.5.1
\ No newline at end of file