|
|
|
|
|
import os |
|
import uuid |
|
import joblib |
|
import json |
|
|
|
import gradio as gr |
|
import pandas as pd |
|
|
|
|
|
|
|
|
|
|
|
os.system("python train.py") |
|
|
|
|
|
|
|
|
|
insurance_charge_predictor = joblib.load("model.joblib") |
|
|
|
|
|
log_file = Path("logs/") / f"data_{uuid.uuid4()}.json" |
|
log_folder = log_file.parent |
|
|
|
scheduler = CommitScheduler( |
|
repo_id="insurance-charge-mlops-logs", |
|
repo_type="dataset", |
|
folder_path=log_folder, |
|
path_in_repo="data", |
|
every=2 |
|
) |
|
|
|
|
|
|
|
def predict_charge(age, sex, bmi, children, somker, region): |
|
smaple = {'age': age, |
|
'bmi': bmi, |
|
'children': children, |
|
'sex': sex, |
|
'smoker': smoker, |
|
'region': region, |
|
'prediction': prediction[0] |
|
} |
|
data_point = pd.DataFrame([sample]) |
|
prediction = insurance_charge_predicter.predict(data_point).tolist() |
|
|
|
|
|
|
|
|
|
|
|
with scheduler.lock: |
|
with log_file.open("a") as f: |
|
f.write(json.dumps( |
|
{ |
|
'age': age, |
|
'bmi': bmi, |
|
'children': children, |
|
'sex': sex, |
|
'smoker': smoker, |
|
'region': region, |
|
'prediction': prediction[0] |
|
} |
|
)) |
|
f.write("\n") |
|
|
|
return prediction[0] |
|
|
|
|
|
|
|
|
|
|
|
age_input = gr.Number(label = "age") |
|
bmi_input = gr.Number(label = "bmi") |
|
children_input = gr.Number(label = "children") |
|
sex_input = gr.Dropdown(["male","female"], label = "sex") |
|
smoker_input = gr.Dropdown(["Yes","No"], label = "smoker") |
|
region_input = gr.Dropdown(["southeast","southwest", "northwest", "northeast"], label = "region") |
|
model_output = gr.Label( label = " Insurance Chaeges") |
|
|
|
|
|
demo = gr.Interface( |
|
fn=predict_insurance_charge, |
|
inputs=[age_input, bmi_input, children_input, sex_input, smoker_input, region_input], |
|
outputs = model_output, |
|
title = "Healthy Insurence Candidate Perdiction", |
|
description = "This API will predict and estimate isurance charges based on candidate's attributes" |
|
examples = [[33,33.44,5,"male","no", "southeast"], |
|
[40,38.20,2,"female","no", "northwest"], |
|
[52,36.20,0,"male","no", "northwest"]], |
|
concurrency_limit = 16 |
|
|
|
) |
|
|
|
|
|
demo.queue() |
|
demo.launch(share=False) |
|
|