Spaces:
Sleeping
Sleeping
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,
|