Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -177,7 +177,7 @@ def plot_plotly(preds: pd.DataFrame):
|
|
| 177 |
# Add touch indicators as points
|
| 178 |
touches = {}
|
| 179 |
for touch_frame in preds.index[preds["Touch"] != ""]:
|
| 180 |
-
teams_players = preds["Touch"]
|
| 181 |
for team, player in zip(teams_players[::2], teams_players[1::2]):
|
| 182 |
team = team.strip()
|
| 183 |
player = player.strip()
|
|
@@ -296,6 +296,7 @@ with TemporaryDirectory() as temp_dir:
|
|
| 296 |
if os.path.exists(preds_file):
|
| 297 |
print(f"Predictions file already exists: {preds_file}")
|
| 298 |
preds = pd.read_csv(preds_file, dtype={"Touch": str})
|
|
|
|
| 299 |
else:
|
| 300 |
preds = infer(MODEL, replay_file,
|
| 301 |
nullify_goal_difference=nullify_goal_difference,
|
|
|
|
| 177 |
# Add touch indicators as points
|
| 178 |
touches = {}
|
| 179 |
for touch_frame in preds.index[preds["Touch"] != ""]:
|
| 180 |
+
teams_players = preds.at[touch_frame, "Touch"].split('|')
|
| 181 |
for team, player in zip(teams_players[::2], teams_players[1::2]):
|
| 182 |
team = team.strip()
|
| 183 |
player = player.strip()
|
|
|
|
| 296 |
if os.path.exists(preds_file):
|
| 297 |
print(f"Predictions file already exists: {preds_file}")
|
| 298 |
preds = pd.read_csv(preds_file, dtype={"Touch": str})
|
| 299 |
+
preds["Touch"] = preds["Touch"].fillna("")
|
| 300 |
else:
|
| 301 |
preds = infer(MODEL, replay_file,
|
| 302 |
nullify_goal_difference=nullify_goal_difference,
|