File size: 885 Bytes
4bf8f67 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import gradio as gr
import joblib
# Chargez votre modèle à partir du fichier pkl
model = joblib.load('best_pipeline.pkl')
# Exemple de données pour un passager
passenger_data = {
"Pclass": "1", # Exemple : "1" pour 1ère classe
"Sex": "male", # Exemple : "male" ou "female"
"Age": 30, # Exemple : 30 ans
"SibSp": 0, # Exemple : 0
"Parch": 0, # Exemple : 0
"Fare": 75.0, # Exemple : 75.0
"Name": "John Smith" # Exemple : Nom du passager
}
def predict_survival(Pclass, Sex, Age, SibSp, Parch, Fare, Name):
# Utilisez votre modèle pour faire des prédictions ici
prediction = model.predict([Pclass, Sex, Age, SibSp, Parch, Fare])
return 'Survécu' if prediction == 1 else 'Non survécu'
iface = gr.Interface(fn=predict_survival, inputs=["text", "text", "number", "number", "number", "number", "text"], outputs="text")
iface.launch()
|