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