diff --git a/train_model.sh b/train_model.sh
index f973634..1e37d51 100755
--- a/train_model.sh
+++ b/train_model.sh
@@ -1,9 +1,10 @@
 #!/bin/bash
 
 BASE_FOLDER=$1
-CONFIG_PATH="./CONFIG.yaml"
+CONFIG_PATH="/app/CONFIG.yaml"
 
 docker run -v $BASE_FOLDER:/data \
+            -v $PWD:/app \
             --gpus all \
-            dcase \
-            poetry run prototypicalbeats/trainer.py fit --config $CONFIG_PATH
\ No newline at end of file
+            beats \
+            poetry run prototypicalbeats/trainer.py fit --config $CONFIG_PATH 
\ No newline at end of file