Spaces:
Sleeping
Sleeping
| import streamlit as st | |
| import joblib | |
| from utils import preprocess_input, predict_stroke | |
| from ui import input_form, display_result | |
| def load_model(path: str = "./models/model.pkl"): | |
| """Load the trained classifier from disk.""" | |
| return joblib.load(path) | |
| def local_css(file_name): | |
| with open(file_name) as f: | |
| st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True) | |
| local_css("styles.css") | |
| def main(): | |
| st.title("Stroke Prediction Demo") | |
| st.write("Enter patient metrics to predict stroke risk/type.") | |
| # Get raw numeric inputs | |
| data = input_form() | |
| # Preprocess and predict | |
| model = load_model() | |
| X = preprocess_input(data) | |
| label, proba = predict_stroke(model, X) | |
| # Show result | |
| display_result(label, proba) | |
| if __name__ == "__main__": | |
| main() | |