GaboDataScientist commited on
Commit
c8f4e87
verified
1 Parent(s): 9540975

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 "The team with higher probability of winning is: " + 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"The team with higher probability of winning is: {Winning_Team} with a probability of {Winning_Proba * 100:.2f}%"
72
 
73
  gui= gr.Interface(
74
  fn=run_my_model,