Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| import pandas as pd | |
| import joblib | |
| # Load the pre-trained model | |
| # Define the input widgets | |
| age = st.slider('Age', 18, 99, 25) | |
| gender = st.selectbox('Gender', ['Male', 'Female']) | |
| smoker = st.selectbox('Smoker', ['Yes', 'No']) | |
| region = st.selectbox('Region', ['Northeast', 'Northwest', 'Southeast', 'Southwest']) | |
| bmi = st.number_input('BMI', min_value=10.0, max_value=50.0, step=0.1) | |
| # Define a function to make the prediction | |
| def predict(age, gender, smoker, region, bmi): | |
| data = pd.DataFrame({'age': [age], | |
| 'sex': [gender], | |
| 'smoker': [smoker], | |
| 'region': [region], | |
| 'bmi': [bmi]}) | |
| prediction = model.predict(data)[0] | |
| return prediction | |
| # Call the predict function and display the result | |
| if st.button('Predict'): | |
| result = predict(age, gender, smoker, region, bmi) | |
| st.write('The predicted insurance cost is $', round(result, 2)) |