Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import pandas as pd | |
| import numpy as np | |
| from joblib import load | |
| def predict_Level( | |
| Age, Gender, Air_Pollution, Alcohol_use, | |
| Dust_Allergy, OccuPational_Hazards, Genetic_Risk, | |
| chronic_Lung_Disease, Balanced_Diet, Obesity, Smoking, | |
| Passive_Smoker, Chest_Pain, Coughing_of_Blood, Fatigue, | |
| Weight_Loss, Shortness_of_Breath, Wheezing,Swallowing_Difficulty, | |
| Clubbing_of_Finger_Nails, Frequent_Cold,Dry_Cough, Snoring | |
| ): | |
| # load model | |
| model = load('Lung_cancer_prediction.joblib') | |
| # Create a dict array from the parameters | |
| data = { | |
| 'Age': [Age], | |
| 'Gender': [Gender], | |
| 'Air Pollution': [Air_Pollution], | |
| 'Alcohol use': [Alcohol_use], | |
| 'Dust Allergy': [Dust_Allergy], | |
| 'OccuPational Hazards': [OccuPational_Hazards], | |
| 'Genetic Risk': [Genetic_Risk], | |
| 'chronic Lung Disease': [chronic_Lung_Disease], | |
| 'Balanced Diet': [Balanced_Diet], | |
| 'Obesity': [Obesity], | |
| 'Smoking': [Smoking], | |
| 'Passive Smoker': [Passive_Smoker], | |
| 'Chest Pain': [Chest_Pain], | |
| 'Coughing of Blood': [Coughing_of_Blood], | |
| 'Fatigue': [Fatigue], | |
| 'Weight Loss': [Weight_Loss], | |
| 'Shortness of Breath': [Shortness_of_Breath], | |
| 'Wheezing': [Wheezing], | |
| 'Swallowing Difficulty': [Swallowing_Difficulty], | |
| 'Clubbing of Finger Nails': [Clubbing_of_Finger_Nails], | |
| 'Frequent Cold': [Frequent_Cold], | |
| 'Dry Cough': [Dry_Cough], | |
| 'Snoring': [Snoring], | |
| } | |
| Xinp = pd.DataFrame(data) | |
| print(Xinp) | |
| # Predict the level | |
| Level = model.predict(Xinp) | |
| # return the level | |
| return Level[0] | |
| # Create the gradio interface | |
| ui = gr.Interface( | |
| fn = predict_Level, | |
| inputs = [ | |
| gr.inputs.Textbox(placeholder='Age', default="33", numeric=True,label='Age'), | |
| gr.inputs.Textbox(placeholder='Gender', default="1", numeric=True,label='Gender'), | |
| gr.inputs.Textbox(placeholder='Air_Pollution', default="2",numeric=True,label='Air Pollution'), | |
| gr.inputs.Textbox(placeholder='Alcohol_use', default="4",numeric=True,label='Alcohol use'), | |
| gr.inputs.Textbox(placeholder='Dust_Allergy', default="5",numeric=True,label='Dust Allergy'), | |
| gr.inputs.Textbox(placeholder='OccuPational_Hazards', default="4",numeric=True,label='OccuPational Hazards'), | |
| gr.inputs.Textbox(placeholder='Genetic_Risk', default="3",numeric=True,label='Genetic Risk'), | |
| gr.inputs.Textbox(placeholder='chronic_Lung_Disease', default="2",numeric=True,label='chronic Lung Disease'), | |
| gr.inputs.Textbox(placeholder='Balanced_Diet', default="2",numeric=True,label='Balanced Diet'), | |
| gr.inputs.Textbox(placeholder='Obesity', default="4", numeric=True,label='Obesity'), | |
| gr.inputs.Textbox(placeholder='Smoking', default="3", numeric=True,label='Smoking'), | |
| gr.inputs.Textbox(placeholder='Passive_Smoker', default="2", numeric=True,label='Passive Smoker'), | |
| gr.inputs.Textbox(placeholder='Chest_Pain', default="2", numeric=True,label='Chest Pain'), | |
| gr.inputs.Textbox(placeholder='Coughing_of_Blood', default="4", numeric=True,label='Coughing of Blood'), | |
| gr.inputs.Textbox(placeholder='Fatigue', default="3",numeric=True,label='Fatigue'), | |
| gr.inputs.Textbox(placeholder='Weight_Loss', default="4", numeric=True,label='Weight Loss'), | |
| gr.inputs.Textbox(placeholder='Shortness_of_Breath', default="2", numeric=True,label='Shortness of Breath'), | |
| gr.inputs.Textbox(placeholder='Wheezing', default="2",numeric=True,label='Wheezing'), | |
| gr.inputs.Textbox(placeholder='Swallowing_Difficulty', default="3", numeric=True,label='Swallowing Difficulty'), | |
| gr.inputs.Textbox(placeholder='Clubbing_of_Finger_Nails', default="1", numeric=True,label='Clubbing of Finger Nails'), | |
| gr.inputs.Textbox(placeholder='Frequent_Cold', default="2", numeric=True,label='Frequent Cold'), | |
| gr.inputs.Textbox(placeholder='Dry_Cough', default="3", numeric=True,label='Dry Cough'), | |
| gr.inputs.Textbox(placeholder='Snoring', default="4", numeric=True,label='Snoring'), | |
| ], | |
| outputs = "text", | |
| ) | |
| if __name__ == "__main__": | |
| ui.launch(share=False) |