gurru16's picture
Upload 3 files
070cff1 verified
Raw
History Blame Contribute Delete
372 Bytes
import gradio as gr
import pickle
with open('titanic_model.pkl', 'rb') as f:
model = pickle.load(f)
def predict(pclass,sex,age,fare):
result = model.predict([[pclass,sex,age,fare]])
return "Survived" if result[0]==1 else "Not Survived"
gr.Interface(
fn=predict,
inputs=["number","number","number","number"],
outputs="text"
).launch()