Added the price preediction output to the predict.py

1 files changed, 1 insertions, 1 deletions

diff --git a/main.py b/main.py
index a8bdec8..b0d6673 100644
--- a/main.py
+++ b/main.py
@@ -90,7 +90,7 @@ def train_model(df, target, model_name):
     # Save the model
     model_file = f'{model_name}.pkl'
     joblib.dump(best_model, model_file)
-    print("Model saved as '{model_file}'")
+    print(f"Model saved as '{model_file}'")
 
 # Main Function
 def main():