Create model_predictor.py

model_predictor.py

@@ -0,0 +1,12 @@
+from joblib import load
+from datetime import datetime, timedelta
+model = load("amc_model.pkl")
+
+def predict_next_service(last_service_date):
+    try:
+        today = datetime.today()
+        delta_days = (today - datetime.strptime(last_service_date, "%Y-%m-%d")).days
+        predicted_days = int(model.predict([[delta_days]])[0])
+        return (today + timedelta(days=predicted_days)).strftime("%Y-%m-%d")
+    except Exception as e:
+        raise ValueError(f"Prediction error: {str(e)}")