GaboDataScientist commited on
Commit
005afbd
verified
1 Parent(s): 9cca7a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -63,10 +63,12 @@ def run_my_model(Home_Team, Away_Team): #aqu铆 ponemos que es solo una entrada p
63
  #Paso7: Creaci贸n del If
64
  if class_names[model.predict(input_df)][0]=="W":
65
  Winning_Team=Home_Team
 
66
  else:
67
  Winning_Team=Away_Team
 
68
 
69
- return "El equipo con mayor probabilidad de ganar es: " + Winning_Team
70
 
71
  gui= gr.Interface(
72
  fn=run_my_model,
 
63
  #Paso7: Creaci贸n del If
64
  if class_names[model.predict(input_df)][0]=="W":
65
  Winning_Team=Home_Team
66
+ Winning_Proba=model.predict_proba(input_df)[0][1]
67
  else:
68
  Winning_Team=Away_Team
69
+ Winning_Proba=model.predict_proba(input_df)[0][0]
70
 
71
+ return f"El equipo con mayor probabilidad de ganar es: {Winning_Team} con una probabilidad de {Winning_Proba * 100:.2f}%"
72
 
73
  gui= gr.Interface(
74
  fn=run_my_model,