Update app.py
Browse files
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:
|
| 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,
|