placementstream / app.py
harshiv's picture
Update app.py
382a39b
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))