Rolv-Arild commited on
Commit
24d25fe
·
verified ·
1 Parent(s): 52dbae6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -116,7 +116,7 @@ def infer(model, replay_file,
116
  preds.index.name = "Frame"
117
  if ignore_ties:
118
  tie_probs = preds["Tie"]
119
- preds = preds.drop("Tie")
120
  preds[[c for c in preds.columns if c.startswith("Blue") or c.startswith("Orange")]] /= (1 - tie_probs)
121
 
122
  return preds
 
116
  preds.index.name = "Frame"
117
  if ignore_ties:
118
  tie_probs = preds["Tie"]
119
+ preds = preds.drop("Tie", axis=1)
120
  preds[[c for c in preds.columns if c.startswith("Blue") or c.startswith("Orange")]] /= (1 - tie_probs)
121
 
122
  return preds