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
|