diff --git a/scripts/training/train.py b/scripts/training/train.py
index e6194ce..0089c09 100644
--- a/scripts/training/train.py
+++ b/scripts/training/train.py
@@ -119,7 +119,7 @@ def train_model():
 
     checkpoint_callback = ModelCheckpoint(
         dirpath="checkpoint_files",
-        filename="inference_recognition_model.ckpt",
+        filename="inference_recognition_model",
         save_top_k=1,
         verbose=True,  # Disable checkpoint verbose output
         monitor="val_acc",