diff --git a/README.md b/README.md
index f14bcf703c2310162fcc3a6b9c13daa4212b6c2d..2c5dbc6b5041c653fcb40603c776ae6ad5c70a2d 100644
--- a/README.md
+++ b/README.md
@@ -20,7 +20,7 @@ Keep in mind that you must download the complete ARC which requires git and git
 
 ### Running deepSTABp
 
-Afterwards you can use the predict_main.py in workflows/TransformerBasedTMPrediction/prediction_model. This means navigate to the respective folder in your command line. Activate the conda environment you created.
+Afterwards you can use the predict_cpu.py/predict.cuda in workflows/prediction_model. This means navigate to the respective folder in your command line. Activate the conda environment you created.
 
 ```
 conda activate environment