Some minor changes

1 files changed, 3 insertions, 1 deletions

diff --git a/main.py b/main.py
index d01d421..6fbb0c2 100644
--- a/main.py
+++ b/main.py
@@ -88,7 +88,7 @@ def train_model(df):
 
     # Save the model
     joblib.dump(best_model, 'model.pkl')
-    print("Model saved as 'best_mileage_predictor.pkl'")
+    print("Model saved as 'model.pkl'")
 
 # Main Function
 def main():
@@ -98,8 +98,10 @@ def main():
 
     df = preprocess_data(df)
     print("Data preprocessing complete.")
+    print("Model is training...")
 
     train_model(df)
+    print("Model is trained.")
 
 if __name__ == "__main__":
     main()