| import streamlit as st | |
| import pandas as pd | |
| import joblib | |
| from huggingface_hub import hf_hub_download | |
| st.set_page_config(page_title="Engine Predictive Maintenance") | |
| st.title("🔧 Engine Predictive Maintenance") | |
| MODEL_REPO = "Vignesh-vigu/engine-predictive-maintenance-model" | |
| MODEL_FILE = "xgboost_model.pkl" | |
| model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE) | |
| model = joblib.load(model_path) | |
| engine_rpm = st.number_input("Engine RPM", 0, 3000, 1000) | |
| lub_oil_pressure = st.number_input("Lub Oil Pressure", value=3.0) | |
| fuel_pressure = st.number_input("Fuel Pressure", value=10.0) | |
| coolant_pressure = st.number_input("Coolant Pressure", value=2.5) | |
| lub_oil_temp = st.number_input("Lub Oil Temperature", value=80.0) | |
| coolant_temp = st.number_input("Coolant Temperature", value=85.0) | |
| input_df = pd.DataFrame([{ | |
| "engine_rpm": engine_rpm, | |
| "lub_oil_pressure": lub_oil_pressure, | |
| "fuel_pressure": fuel_pressure, | |
| "coolant_pressure": coolant_pressure, | |
| "lub_oil_temp": lub_oil_temp, | |
| "coolant_temp": coolant_temp | |
| }]) | |
| if st.button("Predict"): | |
| pred = model.predict(input_df)[0] | |
| if pred == 1: | |
| st.error("⚠️ Engine needs maintenance") | |
| else: | |
| st.success("✅ Engine is healthy") | |