Create model_predictor.py

model_predictor.py

@@ -0,0 +1,11 @@
+from joblib import load
+from datetime import datetime, timedelta
+
+model = load("amc_model.pkl")
+
+def predict_next_service(last_service_date: str) -> str:
+    today = datetime.today()
+    delta_days = (today - datetime.strptime(last_service_date, "%Y-%m-%d")).days
+    predicted_days = int(model.predict([[delta_days]])[0])
+    next_service_date = today + timedelta(days=predicted_days)
+    return next_service_date.strftime("%Y-%m-%d")