johananoa's picture
Add application file
4181ef7
import gradio as gr
import joblib
import numpy as np
def preprocessing(gender):
if gender == "Male":
return 0
elif gender == "Female":
return 1
def predict(Gender,Haemoglobin,MCH,MCHC,MCV):
model = joblib.load("models/svm.pkl")
gender = preprocessing(Gender)
data = np.array([[gender,Haemoglobin,MCH,MCHC,MCV]])
pred = model.predict(data)
if pred == 0:
return "Not Anaemic"
else:
return "Anaemic"
demo = gr.Interface(
fn=predict,
inputs=[gr.Radio(choices=["Male","Female"],value="str",label="Gender",info="Choose Gender"),
gr.Number(label="Haemoglobin"),
gr.Number(label="MCH"),
gr.Number(label="MCHC"),
gr.Number(label="MCV"),
],
outputs=["text"],
)
demo.launch()