init app.py

app.py

@@ -0,0 +1,76 @@
+import gradio as gr
+import numpy as np
+import pickle
+import joblib
+
+def predict(Height,Weight,INR,Race,Age,gender,Diabetes,Simvastatin,Amiodarone,VKORC1, ml_model):
+
+  age_encoder = joblib.load('my_le_age.joblib')
+  en_age = age_encoder.transform([Age])[0]
+  race_encoder = joblib.load('my_le_race.joblib')
+  en_race = race_encoder.transform([Race])[0]
+  gen_encoder = joblib.load('my_le_gender.joblib')
+  en_gen = gen_encoder.transform([gender])[0]
+  sim_encoder = joblib.load('my_le_sim.joblib')
+  en_sim = sim_encoder.transform([Simvastatin])[0]
+  ami_encoder = joblib.load('my_le_ami.joblib')
+  en_ami = ami_encoder.transform([Amiodarone])[0]
+  vk_encoder = joblib.load('my_le_vk.joblib')
+  en_vk = vk_encoder.transform([VKORC1])[0]
+  input_features=np.array([Height,Weight,INR,en_race,en_age,en_gen,Diabetes,en_sim,en_ami,en_vk]).reshape(1,-1)
+  norm_features = scaler.fit_transform(input_features)
+
+  if(ml_model == 'Linear'):
+    with open('lrn_model.pkl', 'rb') as file:
+      lrn_model = pickle.load(file)
+    res = lrn_model.predict(norm_features)[0]
+  elif(ml_model == 'SVRegressor'):
+    with open('svr_model.pkl', 'rb') as file:
+      lrn_model = pickle.load(file)
+    res = lrn_model.predict(norm_features)[0]
+  elif(ml_model == 'Decision Trees Regressor'):
+    with open('dt_model.pkl', 'rb') as file:
+      lrn_model = pickle.load(file)
+    res = lrn_model.predict(norm_features)[0]
+  elif(ml_model == 'Lasso Regression'):
+    with open('lasso_model.pkl', 'rb') as file:
+      lrn_model = pickle.load(file)
+    res = lrn_model.predict(norm_features)[0]
+  else:
+    with open('mlp_model.pkl', 'rb') as file:
+      lrn_model = pickle.load(file)
+    res = lrn_model.predict(norm_features)[0]
+
+  return res
+
+  return en_gen
+
+
+Height = gr.inputs.Number(label='Height (cm)')
+Weight = gr.inputs.Number(label='Weight (kg)')
+INR = gr.inputs.Number(label='INR on Reported Therapeutic Dose of Warfarin')
+Race = gr.inputs.Dropdown(['African-American','Asian','Black','Black African','Black Caribbean',
+ 'Black or African American','Black other','Caucasian','Chinese',
+ 'Han Chinese','Hispanic','Indian','Intermediate' ,'Japanese', 'Korean',
+ 'Malay', 'Other' ,'Other (Black British)' ,'Other (Hungarian)',
+ 'Other Mixed Race' ,'White' ,'other'], label='Race')
+Age = gr.inputs.Dropdown(['10 - 19','20 - 29','30 - 39','40 - 49','50 - 59','60 - 69','70 - 79','80 - 89','90+'],label='Age')
+gender = gr.inputs.Dropdown(["male", "female"], label="Gender")
+Diabetes = gr.inputs.Number(label='Diabetes')
+Simvastatin = gr.inputs.Number(label='Simvastatin (Zocor)')
+Amiodarone = gr.inputs.Number(label='Amiodarone (Cordarone)')
+VKORC1 = gr.inputs.Dropdown(['A/A','A/G','G/G'], label='VKORC1 genotype: -1639 G>A (3673); chr16:31015190; rs9923231; C/T')
+ml_model = gr.inputs.Dropdown(["Linear", "SVRegressor", "Decision Trees Regressor","Lasso Regression", "MLP Neural Network"], label="ML Model")
+outputs = [
+    gr.outputs.Textbox(label="Dose"),
+]
+# Create interface
+interface = gr.Interface(
+    fn=predict,
+    inputs=[Height,Weight,INR,Race,Age,gender,Diabetes,Simvastatin,Amiodarone,VKORC1,
+            ml_model],
+    outputs=outputs
+)
+
+# Launch interface
+interface.launch()