gradio_tutorial / app.py
Rosana's picture
app v1
be367b4
raw
history blame contribute delete
No virus
445 Bytes
import gradio as gr
import joblib as jb
def predict(sex, age, pclass):
model = jb.load('model.pkl')
pclass = int(pclass)
p = model.predict_proba([[sex, age, pclass]])[0]
return {'Did not survive': p[0], 'Survived': p[1]}
demo = gr.Interface(fn = predict,
inputs = [
gr.Dropdown(choices=["male","female"], type="index"),
"number",
gr.Dropdown(choices=["1","2","3"], type="value")
],
outputs="label")
demo.launch()